pretrain false

train_tongue_diagnosis.py

@@ -33,7 +33,7 @@ parser.add_argument('--save_freq', '-s', default=1, type=int, metavar='N', help=
 parser.add_argument('--log_interval', type=int, default=10, metavar='N',
                     help='how many batches to wait before logging training status')
 parser.add_argument('--seed', type=int, default=1, metavar='S', help='random seed (default: 1)')
-parser.add_argument('--pretrain', action='store_true', default=True,
+parser.add_argument('--pretrain', action='store_true', default=False,
                     help='whether to use pretrain')
 
 args = parser.parse_args()