Traceback (most recent call last):
File "train.py", line 344, in
fit_one_epoch(model_train, model, loss, loss_history, optimizer, epoch, epoch_step, epoch_step_val, gen, gen_val, Epoch, Cuda, fp16, scaler, save_period, save_dir, local_rank)
File "/home/stable/dianxuan/code/Siamese-pytorch/utils/utils_fit.py", line 35, in fit_one_epoch
outputs = model_train(images)
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 152, in forward
outputs = self.parallel_apply(replicas, inputs, kwargs)
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 162, in parallel_apply
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply
output.reraise()
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/_utils.py", line 369, in reraise
raise self.exc_type(msg)
ValueError: Caught ValueError in replica 0 on device 0.
Original Traceback (most recent call last):
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 60, in _worker
output = module(*input, **kwargs)
File "/home/stable/anaconda3/envs/sia/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "/home/stable/dianxuan/code/Siamese-pytorch/nets/siamese.py", line 31, in forward
x1, x2 = x
ValueError: not enough values to unpack (expected 2, got 1)
bubbliiiing
