Training Fails

[jetlime]

When executing the proposed training command, the code crashes:

$ python main.py --model-name CNNAutoencoder --device cpu

Training the model!!!
Traceback (most recent call last):
  File "/home/paul/Documents/PANDA/code/main.py", line 179, in <module>
    main(args)
  File "/home/paul/Documents/PANDA/code/main.py", line 163, in main
    trainer(args)
  File "/home/paul/Documents/PANDA/code/train.py", line 151, in trainer
    losses: List = _train_one_epoch(
  File "/home/paul/Documents/PANDA/code/train.py", line 81, in _train_one_epoch
    outputs, tails = model(reshaped_packets)
ValueError: not enough values to unpack (expected 2, got 1)

Environment:

• Python Version: 3.10.14
• OS:

Distributor ID: Ubuntu
Description:    Ubuntu 22.04.4 LTS
Release:        22.04
Codename:       jammy