saikumarkethi

Issue while executing Train.py

10

--------------PyTorch VERSION: 1.7.0+cu101 ..............device cpu Traceback (most recent call last): File "/content/memAE-2/Train.py", line 76, in args.dataset_augment_test_type) File "/content/memAE-2/data/utils.py", line 148, in give_data_folder return train_folder, test_folder UnboundLocalError: local variable 'train_folder' referenced...