Open zhangys-lucky opened 3 years ago
fix code: add callbacks for train fit:
def train(strategy, callbacks=[]):
with strategy.scope():
wide_column, deep_column = columns_builder()
model = create_model(wide_column, deep_column)
train_dataset = read_dataset(TEST_DATA_PATH, C.FILE_MATCH_PATTERN)
model.fit(train_dataset, epochs=1, verbose=1,callbacks=callbacks)
return
train tensornet with callbacks
cp_cb = tn.callbacks.PsWeightCheckpoint("./model")
train(tn.distribute.PsStrategy(), [cp_cb])
保存文档begin-with-wide-deep到
wide_deep.py
执行时报错: