mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-23 02:26:36 +00:00
train is ready
manual_run reuses train's export add current_model to constants
This commit is contained in:
@@ -40,7 +40,6 @@ def export_onnx(model_path, batch_size=4):
|
||||
nms=True,
|
||||
device=0
|
||||
)
|
||||
return onnx_path
|
||||
|
||||
|
||||
def export_tensorrt(model_path):
|
||||
|
||||
Reference in New Issue
Block a user