train is ready

manual_run reuses train's export
add current_model to constants
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 18:41:10 +03:00
parent 0bd5bd6d27
commit 538dc8efa9
4 changed files with 18 additions and 33 deletions
-1
View File
@@ -40,7 +40,6 @@ def export_onnx(model_path, batch_size=4):
nms=True,
device=0
)
return onnx_path
def export_tensorrt(model_path):