xlm998
commented
5 years ago You close this window,then start again,this will run again,try it,
Open WY51BE opened 5 years ago
xlm998
commented
5 years ago You close this window,then start again,this will run again,try it,
There is this problem when running,I have run without problems before.Suddenly it not work. Can you help me？
Traceback (most recent call last): File "demo.py", line 35, in
config = cfg.model.m2det_config)
File "/media/dat1/users/master/2019/wuyi/M2Det-master-change/m2det.py", line 194, in build_net
return M2Det(phase, size, config)
File "/media/dat1/users/master/2019/wuyi/M2Det-master-change/m2det.py", line 36, in init
self.construct_modules()
File "/media/dat1/users/master/2019/wuyi/M2Det-master-change/m2det.py", line 54, in construct_modules
side_channel=512)) #side channel isn't fixed.
File "/media/dat1/users/master/2019/wuyi/M2Det-master-change/layers/nn_utils.py", line 38, in init
self.layers.add_module('{}'.format(len(self.layers)), BasicConv(self.in1, self.planes, 3, 2, 1))
File "/media/dat1/users/master/2019/wuyi/M2Det-master-change/layers/nn_utils.py", line 14, in init
stride=stride, padding=padding, dilation=dilation, groups=groups, bias=bias)
File "/media/dat1/users/master/2019/wuyi/anaconda3/envs/M2Det/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 315, in init
False, _pair(0), groups, bias)
File "/media/dat1/users/master/2019/wuyi/anaconda3/envs/M2Det/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 38, in init
out_channels, in_channels // groups, *kernel_size))
RuntimeError: CUDA error: out of memory