Traceback (most recent call last):


 "./pytorch/train.py", line 674, in <module>

fire.Fire()


 "/opt/conda/lib/python3.6/site-packages/fire/core.py", line 138, in Fire

component_trace = _Fire(component, args, parsed_flag_args, context, name)


 "/opt/conda/lib/python3.6/site-packages/fire/core.py", line 468, in _Fire

target=component.__name__)


 "/opt/conda/lib/python3.6/site-packages/fire/core.py", line 672, in _CallAndUpdateTrace

component = fn(*varargs, **kwargs)


 "./pytorch/train.py", line 490, in evaluate

torchplus.train.try_restore_latest_checkpoints(model_dir, [net])


 "/workspace/second.pytorch/torchplus/train/checkpoint.py", line 149, in try_restore_latest_checkpoints

restore(latest_ckpt, model, map_func)


 "/workspace/second.pytorch/torchplus/train/checkpoint.py", line 120, in restore

model.load_state_dict(state_dict)


 "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1049, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for VoxelNet:

Traceback (most recent call last):


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

main()


 "main.py", line 152, in main

loss = train(imgL_crop,imgR_crop, disp_crop_L)


 "main.py", line 95, in train

output1, output2, output3 = model(imgL,imgR)


 "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/usr/local/lib/python2.7/dist-packages/torch/nn/parallel/data_parallel.py", line 114, in forward

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


 "/usr/local/lib/python2.7/dist-packages/torch/nn/parallel/data_parallel.py", line 124, in parallel_apply

return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])


 "/usr/local/lib/python2.7/dist-packages/torch/nn/parallel/parallel_apply.py", line 65, in parallel_apply

raise output
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/aten/src/THC/generic/THCStorage.cu:58
