Traceback (most recent call last):


 "main.py", line 437, in <module>

main()


 "main.py", line 121, in main

main_worker(args.gpu, ngpus_per_node, args)


 "main.py", line 257, in main_worker

train(train_loader, model, criterion, optimizer, epoch, args)


 "main.py", line 299, in train

output = model(input)


 "<myenv>/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__

result = self.forward(*input, **kwargs)


 "<myenv>/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 153, in forward

return self.gather(outputs, self.output_device)


 "<myenv>/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 165, in gather

return gather(outputs, output_device, dim=self.dim)


 "<myenv>/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 68, in gather

res = gather_map(outputs)


 "<myenv>/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 63, in gather_map

return type(out)(map(gather_map, zip(*outputs)))
TypeError: __new__() missing 2 required positional arguments: 'aux_logits2' and 'aux_logits1'

`Traceback (most recent call last):


 "main.py", line 301, in <module>

main()


 "main.py", line 124, in main

train(train_loader, model, criterion, optimizer, epoch)


 "main.py", line 166, in train

output = model(input_var)


 "/home/zouyiping/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__

result = self.forward(*input, **kwargs)


 "/home/zouyiping/.local/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 152, in forward

outputs = self.parallel_apply(replicas, inputs, kwargs)


 "/home/zouyiping/.local/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)])


 "/home/zouyiping/.local/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 85, in parallel_apply

output.reraise()


 "/home/zouyiping/.local/lib/python3.6/site-packages/torch/_utils.py", line 385, in reraise

raise self.exc_type(msg)
TypeError: Caught TypeError in replica 0 on device 0.
