When I tried to train vq_vae, the program was stuck on line 52 of train_vq.py which is for log printing. And I added "print(1)" after line 52, the "1" cannot be printed. Can you help me with some advice?
I checked again and found that the program was stuck on line 21 of evaluator_wrapper.py:
checkpoint = torch.load(pjoin(opt.checkpoints_dir, opt.dataset_name, 'text_mot_match', 'model', 'finest.tar'), map_location=opt.device)
How can I fix this?
When I tried to train vq_vae, the program was stuck on line 52 of train_vq.py which is for log printing. And I added "print(1)" after line 52, the "1" cannot be printed. Can you help me with some advice?
I checked again and found that the program was stuck on line 21 of evaluator_wrapper.py: checkpoint = torch.load(pjoin(opt.checkpoints_dir, opt.dataset_name, 'text_mot_match', 'model', 'finest.tar'), map_location=opt.device) How can I fix this?