Skip to content

Commit 21226ff

Browse files
[Bug-fix] Remove extra directory created at checkpoint (#4675) (#4678)
* [Bug-fix] Remove extra directory created at checkpoint * removing os import
1 parent da699f4 commit 21226ff

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

ml-agents/mlagents/trainers/torch/model_serialization.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import os
21
import threading
32
from mlagents.torch_utils import torch
43

@@ -90,9 +89,6 @@ def export_policy_model(self, output_filepath: str) -> None:
9089
9190
:param output_filepath: file path to output the model (without file suffix)
9291
"""
93-
if not os.path.exists(output_filepath):
94-
os.makedirs(output_filepath)
95-
9692
onnx_output_path = f"{output_filepath}.onnx"
9793
logger.info(f"Converting to {onnx_output_path}")
9894

0 commit comments

Comments
 (0)