Skip to content
Snippets Groups Projects
Commit da87e514 authored by hemaohua's avatar hemaohua
Browse files

delete unnecessary logs

parent 7af7c741
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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment