djiajunustc / TransVG

157 stars 26 forks source link

Training with refcoco+ #14

Closed pqviet closed 2 years ago

pqviet commented 2 years ago

I have finished training with 3 datasets (referit, refcoco, refcocog). Traning with refcoco+ failed with following error messages

File "train.py", line 284, in main(args) File "train.py", line 193, in main dataset_train = build_dataset('train', args) File "TransVG/datasets/init.py", line 67, in build_dataset max_query_len=args.max_query_len) File "TransVG/datasets/data_loader.py", line 213, in init self.images += torch.load(imgset_path) File "/opt/conda/lib/python3.7/site-packages/torch/serialization.py", line 585, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/opt/conda/lib/python3.7/site-packages/torch/serialization.py", line 765, in _legacy_load result = unpickler.load() _pickle.UnpicklingError: invalid load key, '\x00'.

The problem occurred when it tried to load 'data/unc+/unc+_train.pth'. I re-downloaded data.tar, but still cound not fix it. Can you confirm this problem?