Skip to content
Snippets Groups Projects
Unverified Commit 2cbb97c6 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3973 delete unnecessary logs

Merge pull request !3973 from hemaohua/master
parents 74bcf957 da87e514
No related branches found
No related tags found
No related merge requests found
......@@ -13,12 +13,6 @@
# limitations under the License.
# ============================================================================
"""train callbacks"""
try:
from moxing.framework import file
print("import moxing success")
except ModuleNotFoundError as e:
print(f'not modelarts env, error={e}')
import os
import time
......@@ -132,6 +126,7 @@ class Val_Callback(Callback):
ckpt_file = os.path.join(ckpt_path, 'model_checkpoint.ckpt')
save_checkpoint(cb_params.train_network, ckpt_file)
if self.checkpoint_path.startswith('s3://') or self.checkpoint_path.startswith('obs://'):
from moxing.framework import file
file.copy_parallel(save_path, os.path.join(
self.checkpoint_path, model_info))
print('==============save checkpoint finished===================')
......
......@@ -13,12 +13,6 @@
# limitations under the License.
# ============================================================================
"""train the PDarts model"""
try:
from moxing.framework import file
print("import moxing success")
except ModuleNotFoundError as e:
print(f'not modelarts env, error={e}')
import os
import time
import logging
......@@ -136,6 +130,7 @@ def main():
load_param_into_net(network, param_dict)
if args.data_url.startswith('s3://') or args.data_url.startswith('obs://'):
from moxing.framework import file
data_url_cache = os.path.join(args.local_data_root, 'data')
file.copy_parallel(args.data_url, data_url_cache)
args.data_url = data_url_cache
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment