Closed takeyamayuki closed 1 year ago
I have corrected the following, but got another error.
(morig) yonelab@yonePC:~/workspace/MoRig$ python -u training/train_rig.py --arch="jointnet_motion" -e --resume="checkpoints/jointnet_motion/model_best.pth.tar" --train_folder="ModelsResources/train/" --val_folder="ModelsResources/val/" --test_folder="ModelsResources/test/" --output_folder="results/our_results"
Namespace(aggr_method='attn', arch='jointnet_motion', checkpoint='checkpoints/test', epochs=120, evaluate=True, gamma=0.2, logdir='logs/test', lr=0.0005, motion_dim=32, num_keyframes=5, output_folder='results/our_results', resume='checkpoints/jointnet_motion/model_best.pth.tar', schedule=[40, 80], start_epoch=0, test_batch=2, test_folder='ModelsResources/test/', train_batch=2, train_folder='ModelsResources/train/', val_folder='ModelsResources/val/', weight_decay=0.0001)
=> loading checkpoint 'checkpoints/jointnet_motion/model_best.pth.tar'
=> loaded checkpoint 'checkpoints/jointnet_motion/model_best.pth.tar' (epoch 108)
Total params: 7.88M
Processing...
0%| | 0/2163 [00:00<?, ?it/s]
Traceback (most recent call last):
File "training/train_rig.py", line 294, in <module>
main(parser.parse_args())
File "training/train_rig.py", line 103, in main
train_loader = DataLoader(RigDataset(root=args.train_folder), batch_size=args.train_batch, shuffle=True, follow_batch=['joints'])
File "./datasets/dataset_rig.py", line 13, in __init__
super(RigDataset, self).__init__(root)
File "/home/yonelab/anaconda3/envs/morig/lib/python3.7/site-packages/torch_geometric/data/in_memory_dataset.py", line 56, in __init__
super().__init__(root, transform, pre_transform, pre_filter)
File "/home/yonelab/anaconda3/envs/morig/lib/python3.7/site-packages/torch_geometric/data/dataset.py", line 87, in __init__
self._process()
File "/home/yonelab/anaconda3/envs/morig/lib/python3.7/site-packages/torch_geometric/data/dataset.py", line 170, in _process
self.process()
File "./datasets/dataset_rig.py", line 83, in process
v_traj = np.load(vtx_filename)
File "/home/yonelab/anaconda3/envs/morig/lib/python3.7/site-packages/numpy/lib/npyio.py", line 417, in load
fid = stack.enter_context(open(os_fspath(file), "rb"))
FileNotFoundError: [Errno 2] No such file or directory: 'ModelsResources/train/raw/ModelsResources/train/540_vtx_traj.npy'
but sorry, the npy file did not exist. I'll try a few more things.
I'll close this issue now that it's resolved for now.
Thanks for the great project.
After following the README and building the environment, I ran it and got the following error.
And here is my environment Ubuntu20.04, Python 3.7.13
How can I deal with this? Thank you in advance.