code_5.py 114 字节
Newer Older
ToTensor's avatar
ToTensor 已提交
1 2
callbacks = [earlystop, reducelr, modelckpt] # 设定回调
history = ann.fit(X_train, y_train, # 指定训练集