Closed ghost closed 3 years ago
Hi,
You could try batch size to 2 if it keeps giving you out of memeory error.
It was not about batch size. at least in my system, when I use your comment for testing: 'CUDA_VISIBLE_DEVICES=0,1 python train_detector_se.py -b 1 --resume ‘best_model.ckpt’ --test 1 --save-dir /output/' it gives the memory error I mentioned. I changed this part of your code and it worked. parser.add_argument('--gpu', default='0,1', type=str, metavar='N', help='use gpu')
Hello, I would like to first thank you for this nice project. I have some questions about your model: