ValueError: too many values to unpack (expected 4) - Training the backbone model.

[navarmn]

Epoch: [1][20577/20578] Time 0.761 Data 0.001 Loss 136.7847 Epoch: [1][20578/20578] Time 0.761 Data 0.001 Loss 157.6949 Traceback (most recent call last): File "train.py", line 401, in trainer.run(train_loader, args.config.training.num_epochs) File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 326, in run File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 291, in _handle_exception File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 317, in run File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 226, in _fire_event File "train.py", line 269, in log_epoch evaluator.run(train_loader) File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 326, in run File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 291, in _handle_exception File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 313, in run File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 280, in _run_once_on_dataset File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 291, in _handle_exception File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 273, in _run_once_on_dataset File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/engine/engine.py", line 226, in _fire_event File "/usr/local/lib/python3.6/dist-packages/torch/autograd/grad_mode.py", line 43, in decorate_no_grad return func(*args, **kwargs) File "/usr/local/lib/python3.6/dist-packages/pytorch_ignite-0.1.2-py3.6.egg/ignite/metrics/metric.py", line 65, in iteration_completed File "/home/navar/savoz/aes-lac-2018/codes/metrics.py", line 17, in update self.metrics[i].update(output) File "/home/navar/savoz/aes-lac-2018/codes/metrics.py", line 46, in update out, targets, out_sizes, target_sizes = output ValueError: too many values to unpack (expected 4)

[navarmn]

A turn-around, not a solution: I just realized the --checkpoint argument is not working. I am using --checkpoint-per-batch instead.

The model saves fine at the end of each batch, and the code brokes at the end of the training.