GPU Memory use too much? #5
Description
I run the test code ‘CUDA_VISIBLE_DEVICES=0,1,2,3 python cifar10_defense.py --fname test --md_path /local/rcs/mcz/2021Spring/RobPretrained/unlabeled-rob/cifar10_rst_adv.pt.ckpt --carmon --eval_only --ssl_model_path /local/rcs/mcz/2021Spring/SSRobdata/unlab_cifar10_srn28-10_carmon/March1/ssl_model_130.pth’ in a machine with 4x4090,it still occur torch.cuda.OutOfMemoryError: CUDA out of memory.