-
Notifications
You must be signed in to change notification settings - Fork 40
Open
Description
When I try to execute train.py, I met this error.
-------YOUR SETTINGS_________
verbose: True
model: VRES
train_set: train
fine_tune: False
num_epochs: 50
batch_size: 100
learning_rate: 0.001
scale: 3
Contructing dataset...
Training...
Traceback (most recent call last):
File "train.py", line 76, in
main()
File "train.py", line 74, in main
solver.train(train_dataset)
File "C:\Users\csy\Documents\video-super-resolution-master\solver.py", line 233, in train
self._epoch_step(train_dataset, epoch)
File "C:\Users\csy\Documents\video-super-resolution-master\solver.py", line 101, in _epoch_step
running_loss += loss.data[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels