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 @@ ...@@ -13,12 +13,6 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================
"""train callbacks""" """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 os
import time import time
...@@ -132,6 +126,7 @@ class Val_Callback(Callback): ...@@ -132,6 +126,7 @@ class Val_Callback(Callback):
ckpt_file = os.path.join(ckpt_path, 'model_checkpoint.ckpt') ckpt_file = os.path.join(ckpt_path, 'model_checkpoint.ckpt')
save_checkpoint(cb_params.train_network, ckpt_file) save_checkpoint(cb_params.train_network, ckpt_file)
if self.checkpoint_path.startswith('s3://') or self.checkpoint_path.startswith('obs://'): 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( file.copy_parallel(save_path, os.path.join(
self.checkpoint_path, model_info)) self.checkpoint_path, model_info))
print('==============save checkpoint finished===================') print('==============save checkpoint finished===================')
......
...@@ -13,12 +13,6 @@ ...@@ -13,12 +13,6 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================
"""train the PDarts model""" """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 os
import time import time
import logging import logging
...@@ -136,6 +130,7 @@ def main(): ...@@ -136,6 +130,7 @@ def main():
load_param_into_net(network, param_dict) load_param_into_net(network, param_dict)
if args.data_url.startswith('s3://') or args.data_url.startswith('obs://'): 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') data_url_cache = os.path.join(args.local_data_root, 'data')
file.copy_parallel(args.data_url, data_url_cache) file.copy_parallel(args.data_url, data_url_cache)
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