mirror/private-detector
1
0
Fork 0
mirror of https://github.com/bumble-tech/private-detector.git synced 2026-01-06 04:09:54 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
Gleb Vazhenin
2023-11-05 22:14:36 +00:00
committed by GitHub
parent 33a87bf1d0
commit 3a5c81e514
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -214,7 +214,7 @@ def train(train_id: str,
logger.info(f'Training complete, saving model to {train_id}')
model.save(train_id)
model.save_base64_serving(train_id, input_dtype=dtype)
model.save_base64_serving(f'{train_id}_base64', input_dtype=dtype)
logger.info(f'Model saved to {train_id}')