显存问题

[xiaohongzaizhe]

我使用了48G的显存的显卡，parser.add_argument('-chunk', type=int, default=5 , help='crop volume depth') #原本96 parser.add_argument('-num_sample', type=int, default=1 , help='sample pos and neg') parser.add_argument('-roi_size', type=int, default=96 , help='resolution of roi') parser.add_argument('-b', type=int, default=1, help='batch size for dataloader')都会爆显存，而太低的参数会导致loss=1 模式崩溃