diff --git a/exp/kerokero/train.py b/exp/kerokero/train.py --- a/exp/kerokero/train.py +++ b/exp/kerokero/train.py @@ -1,14 +1,16 @@ +import os +from time import time import argparse import logging as log import numpy as np from keras.layers import Conv2D,Dropout,Dense,Flatten,MaxPooling2D,BatchNormalization from keras.models import Sequential,load_model +from keras.callbacks import TensorBoard +import config as cfg import ftp -log.basicConfig(level=log.INFO,format="%(asctime)s %(levelname)s: %(message)s") - parser=argparse.ArgumentParser() parser.add_argument("data") parser.add_argument("--load_model") @@ -82,10 +84,11 @@ with np.load(args.data) as data: testLabels=data["testLabels"] 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) + 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) log.info("saving model...") model.save(path) - if i%2==1: ftp.push(path) -log.info(model.evaluate(testImages,testLabels)) + ftp.push(path) +log.info(model.evaluate(testImages.reshape((-1,224,224,1)),testLabels))