训练Omniglot数据集时出现这个报错，ValueError: not enough values to unpack (expected 2, got 1)

loongscale commented 1 year ago

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)