CUDA_VISIBLE_DEVICES=3 python3 train.py --mode teacher #8
Description
CUDA_VISIBLE_DEVICES=3 python3 train.py --mode teacher
When I execute the above command
I encountered the following errors
Traceback (most recent call last):
File "train.py", line 243, in
train()
File "train.py", line 198, in train
loss_l, loss_c, loss_landm, pos, neg = criterion(out, priors, targets)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/app/home/houcaoya/code/face_recognition/TDKD/layers/modules/multibox_loss.py", line 57, in forward
loc_data, conf_data, landm_data, _, _, _, _, _ = predictions
ValueError: not enough values to unpack (expected 8, got 3)