Update train.py

pynicolas · web-flow · commit cba7c139c954 · 2025-06-20T22:14:56.000+02:00
diff --git a/train.py b/train.py
@@ -226,4 +226,4 @@ def evaluate_encoder(encoder_name, model_save_path, device=torch.device('cpu')):
   quant_config=quant_config
 )
 edge_model_quantized.export(TFLITE_MODEL_FILE_PATH)
-print(f"Wrote {TFLITE_MODEL_FILE_PATH}")
+print(f"Wrote TFLite model: {TFLITE_MODEL_FILE_PATH}")

Original file line number	Diff line number	Diff line change
`@@ -226,4 +226,4 @@ def evaluate_encoder(encoder_name, model_save_path, device=torch.device('cpu')):`
`226`	`226`	`quant_config=quant_config`
`227`	`227`	`)`
`228`	`228`	`edge_model_quantized.export(TFLITE_MODEL_FILE_PATH)`
`229`		`-print(f"Wrote {TFLITE_MODEL_FILE_PATH}")`
	`229`	`+print(f"Wrote TFLite model: {TFLITE_MODEL_FILE_PATH}")`