Skip to content

Commit 03e4c59

Browse files
committed
change save_strategy to "epoch"
1 parent 2963a4c commit 03e4c59

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autointent/_wrappers/embedder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def train(self, utterances: list[str], labels: list[int], config: EmbedderFineTu
143143
loss = BatchAllTripletLoss(model=self.embedding_model, margin=config.margin)
144144
with tempfile.TemporaryDirectory() as tmp_dir:
145145
args = SentenceTransformerTrainingArguments(
146-
save_strategy="no",
146+
save_strategy="epoch",
147147
output_dir=tmp_dir,
148148
num_train_epochs=config.epoch_num,
149149
per_device_train_batch_size=config.batch_size,

0 commit comments

Comments
 (0)