`Traceback (most recent call last):


 "train.py", line 257, in <module>

train()


 "train.py", line 97, in train

ssd_net = build_ssd('train', cfg['min_dim'], cfg['num_classes'])


 "/media/disk/wudi/ssd.pytorch/ssd.py", line 207, in build_ssd

base_, extras_, head_ = multibox(vgg(base[str(size)], 3),


 "/media/disk/wudi/ssd.pytorch/ssd.py", line 136, in vgg

conv2d = nn.Conv2d(in_channels, v, kernel_size=3, padding=1)


 "/media/disk/wudi/.local/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 330, in __init__

False, _pair(0), groups, bias, padding_mode)


 "/media/disk/wudi/.local/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 41, in __init__

out_channels, in_channels // groups, *kernel_size))
RuntimeError: CUDA error: out of memory

`Traceback (most recent call last):


 "TestNet.py", line 285, in <module>

trainer.fit(model)


 /path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1003, in fit

results = self.single_gpu_train(model)


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/distrib_parts.py", line 186, in single_gpu_train

results = self.run_pretrain_routine(model)


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1213, in run_pretrain_routine

self.train()


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/training_loop.py", line 370, in train

self.run_training_epoch()


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/training_loop.py", line 452, in run_training_epoch

batch_output = self.run_training_batch(batch, batch_idx)


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/training_loop.py", line 632, in run_training_batch

self.hiddens


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/training_loop.py", line 783, in optimizer_closure

training_step_output = self.process_output(training_step_output, train=True)


 "/path/to/pytorch/lib64/python3.6/site-packages/pytorch_lightning/trainer/logging.py", line 159, in process_output

'No `loss` value in the dictionary returned from `model.training_step()`.'
RuntimeError: No `loss` value in the dictionary returned from `model.training_step()`.
