diff --git a/exp/kerokero/train.py b/exp/kerokero/train.py --- a/exp/kerokero/train.py +++ b/exp/kerokero/train.py @@ -85,10 +85,11 @@ with np.load(args.data) as data: log.info("done") tensorboard = TensorBoard(log_dir=os.path.join(cfg.thisDir,"../logs","{}".format(time()))) -for i in range(args.initial_epoch//10,args.epochs//10): - model.fit(trainImages.reshape((-1,224,224,1)),trainLabels,epochs=(i+1)*10,initial_epoch=i*10,batch_size=128,validation_split=0.2,callbacks=[tensorboard]) - path=args.save_model.format((i+1)*10) +BIG_STEP=50 +for i in range(args.initial_epoch//BIG_STEP,args.epochs//BIG_STEP): + model.fit(trainImages.reshape((-1,224,224,1)),trainLabels,epochs=(i+1)*BIG_STEP,initial_epoch=i*BIG_STEP,batch_size=128,validation_split=0.2,callbacks=[tensorboard]) + path=args.save_model.format((i+1)*BIG_STEP) log.info("saving model...") model.save(path) - ftp.push(path) + # ftp.push(path) log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels))