cuda error: out of memory about medicalnet HOT 3 OPEN

you can decrease the input size in setting.py file.Like I changed to 969696 from 44844850
parser.add_argument(
'--input_D',
default=96,
type=int,
help='Input size of depth')
parser.add_argument(
'--input_H',
default=96,
type=int,
help='Input size of height')
parser.add_argument(
'--input_W',
default=96,
type=int,
help='Input size of width')

from medicalnet.

Same problem with 6g memories.

from medicalnet.

thx, it works
BUT does the size have any influence on the result？

from medicalnet.