minor change

train_tongue_diagnosis.py

@@ -32,7 +32,7 @@ parser.add_argument('--log_interval', type=int, default=10, metavar='N',
 
 args = parser.parse_args()
 
-NUM_CLASSES = 7
+NUM_CLASSES = 8
 use_cuda = not args.no_cuda and torch.cuda.is_available()
 GPUID = args.gpuid
 os.environ["CUDA_VISIBLE_DEVICES"] = str(GPUID)
@@ -43,6 +43,8 @@ model_dir = args.model_dir
 if not os.path.exists(model_dir):
     os.makedirs(model_dir)
 
+torch.backends.cudnn.enabled = False
+
 # training data and test data settings
 kwargs = {'num_workers': 4, 'pin_memory': True} if use_cuda else {}
 normalizer = transforms.Normalize(mean=[x / 255.0 for x in [125.3, 123.0, 113.9]],