diff --git a/exp/kerokero/train.py b/exp/kerokero/train.py --- a/exp/kerokero/train.py +++ b/exp/kerokero/train.py @@ -82,10 +82,10 @@ with np.load(args.data) as data: testLabels=data["testLabels"] log.info("done") -for i in range(args.initial_epoch,args.epochs//10): +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) - path=args.save_model.format(i+1) + path=args.save_model.format((i+1)*10) log.info("saving model...") model.save(path) - ftp.push(path) + if i%2==1: ftp.push(path) log.info(model.evaluate(testImages,testLabels))