Description
when I train my code,when it comes to multi_gpu_test part, memory error occur
collect_results_cpu
mmcv.dump(result_part, osp.join(tmpdir, 'part_{}.pkl'.format(rank)))
File "/home/anaconda2/envs/lib/python3.6/site-packages/mmcv/fileio/io.py", line 80, in dump
handler.dump_to_path(obj, file, **kwargs)
File "/home/anaconda2/envs/lib/python3.6/site-packages/mmcv/fileio/handlers/pickle_handler.py", line 26, in dump_to_path
obj, filepath, mode='wb', **kwargs)
File "/home/anaconda2/envs/lib/python3.6/site-packages/mmcv/fileio/handlers/base.py", line 27, in dump_to_path
self.dump_to_fileobj(obj, f, **kwargs)
File "/home/anaconda2/envs/lib/python3.6/site-packages/mmcv/fileio/handlers/pickle_handler.py", line 22, in dump_to_fileobj
pickle.dump(obj, file, **kwargs)
MemoryError
any suggestion for the problem?
Activity