import torch from pathlib import Path dataset_test = torch.load(Path("/media/wombat/c6928dc9-ba03-411d-9483-8e28df5973b9/Music Data/HuggingFace_Mistral_Transformer_Single_Instrument/data/dataset_valid.pt"), weights_only=False) print(f"valid dataset length: {len(dataset_test)}")