-
Notifications
You must be signed in to change notification settings - Fork 249
Description
Namespace(batch_size=25, epochs=500, evaluate=False, lr=0.0005, resume='', start_epoch=0)
==> (Training video, Validation video):( 9537 3783 )
Eligible videos for training : 9492 videos
Eligible videos for validation: 3758 videos
==> Build model and setup loss and optimizer
==> Epoch:[0/500][training stage]
0%| | 0/380 [00:10<?, ?it/s]
Traceback (most recent call last):
File "spatial_cnn.py", line 263, in
main()
File "spatial_cnn.py", line 61, in main
model.run()
File "spatial_cnn.py", line 113, in run
self.train_1epoch()
File "spatial_cnn.py", line 154, in train_1epoch
output = Variable(torch.zeros(len(data_dict['img1']),101).float()).cuda()
TypeError: new(): invalid data type 'str'
when run spatial_cnn.py,I meet this,can anyone help me?