Browse Source

enable_modelarts

master
liupengfei 1 month ago
parent
commit
f53016cd45
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      default_config.yaml
  2. +1
    -1
      src/model_utils/moxing_adapter.py

+ 1
- 1
default_config.yaml View File

@@ -48,7 +48,7 @@ mindir_file_name: "Flownet2" # Save file path
file_format: "MINDIR" # Save file format

# Modelarts Setup
enable_modelarts: 0 # Is training on modelarts
enable_modelarts: 1 # Is training on modelarts


# data_url,train_url是固定用于在modelarts上训练的参数,表示数据集的路径和输出模型的路径


+ 1
- 1
src/model_utils/moxing_adapter.py View File

@@ -18,7 +18,7 @@
import os
import functools
from mindspore import context
from config import config
from .config import config

_global_sync_count = 0



Loading…
Cancel
Save