Revise save_dir

This commit is contained in:
linyiLYi 2023-04-27 09:56:45 +08:00
parent 3bb2933751
commit 78cb661e6c

View File

@ -86,7 +86,7 @@ def main():
)
# Set the save directory
save_dir = "trained_models_test"
save_dir = "trained_models"
os.makedirs(save_dir, exist_ok=True)
# Load the model from file