Pytorch0.4.1_Openpose
Pytorch0.4.1_Openpose copied to clipboard
train problem
When i run 'python train.py' to 50%, here is the error:
loading annotations into memory...
Done (t=4.52s)
creating index...
index created!
loading annotations into memory...
Done (t=0.53s)
creating index...
index created!
train images: 64115
val images: 2693
50%|████████████████▉ | 8014/16029 [3:14:17<3:02:36, 1.37s/it]Traceback (most recent call last):
File "train.py", line 16, in
before 50% is ok,but why need this '.pth' file? anyone help?
I have the same problem! anyone help?
@JiangDing1995 @Johnren1111
- create a new file named "model" in work_space,
- modify all code about file path, such as:
params['work_space']/'save' ->
os.path.join(params['work_space'],'save')
save_path / ('model_{}_val_loss:{}_step:{}.pth'.format(time, val_loss, self.step) ->
os.path.join(save_path,'model_{}val_loss{}step{}.pth'.format(time, val_loss, self.step)