Skip to content

Error in network training #7

@caisy417

Description

@caisy417

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions