search

ChenWWWeixiang / OpenCovidDetector

New virsion for multi-categories were available. Since too many changes above old virsion, please refer to this site https://github.com/ChenWWWeixiang/diagnosis_covid19
18 stars 6 forks source link

GPU memory requirements #4

Closed MadisLemsalu closed 4 years ago

MadisLemsalu commented 4 years ago

Which GPU did you use for training? Did you downsample the data?

ChenWWWeixiang commented 4 years ago

We used 2080Ti 11G to train our network. Since training is done on slice level, slices were all resized to 256x256 and training batch size= 32