Skip to content

Commit df58e7c

Browse files
committed
include configs save in training of transformer
1 parent c260b39 commit df58e7c

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed
Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,3 @@
1-
beautifulsoup4
1+
beautifulsoup4
2+
tf2onnx==1.14.0
3+
onnx==1.12.0

Tutorials/09_translation_transformer/train.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,8 @@ def preprocess_inputs(data_batch, label_batch):
123123
model2onnx = Model2onnx(f"{configs.model_path}/model.h5", metadata={"tokenizer": tokenizer.dict(), "detokenizer": detokenizer.dict()}, save_on_epoch_end=False)
124124
encDecSplitCallback = EncDecSplitCallback(configs.model_path, encoder_metadata={"tokenizer": tokenizer.dict()}, decoder_metadata={"detokenizer": detokenizer.dict()})
125125

126+
configs.save()
127+
126128
# Train the model
127129
transformer.fit(
128130
train_dataProvider,

0 commit comments

Comments
 (0)