def main(_): #mode = "evalfreeze" mode = "pretrain" evaldata = "20171213142927.xls" model = DomainNameModel(mode='freeze', learning_rate=0.0008) worker = Work(model, batch_size=1024, pretrain_iter=20000, train_iter=900000, sample_iter=100, model_save_path='model', sample_path='sample', plk_dir='./plk') if mode == 'pretrain': pretain() else: #eval dosample = preprosample(None, plk_dir='./plk/') worker.evalfreeze(evaldata, dosample)