diff --git a/main_custom.py b/main_custom.py index fb68fa6e6facf3798ad984c373cdc8d0cc2549f8..283c0fccb065e8989ccb6d79b03d084be6494003 100644 --- a/main_custom.py +++ b/main_custom.py @@ -219,7 +219,7 @@ def main(args): batch_size=args.batch_size, length=25) elif args.forward == 'transfer': - data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,'database/data_holdout.cvs','database/data_holdout.cvs'], + data_train, _, _ = dataloader.load_data(data_sources=[args.dataset_train,'database/data_holdout.csv','database/data_holdout.csv'], batch_size=args.batch_size, length=25) _, data_val, data_test = common_dataset.load_data(path_train=args.dataset_val,