File tree Expand file tree Collapse file tree 3 files changed +10
-3
lines changed Expand file tree Collapse file tree 3 files changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -51,7 +51,9 @@ def __init__(self, config):
51
51
self .global_config = config .Global
52
52
self .eval_config = config .Evaluate
53
53
54
- config_path = self .get_config_path (self .eval_config .weight_path )
54
+ config_path = self .eval_config .get ("basic_config_path" , None )
55
+ if not config_path :
56
+ config_path = self .get_config_path (self .eval_config .weight_path )
55
57
56
58
self .pdx_config , self .pdx_model = build_model (
57
59
self .global_config .model , config_path = config_path
Original file line number Diff line number Diff line change @@ -51,7 +51,9 @@ def __init__(self, config):
51
51
self .global_config = config .Global
52
52
self .export_config = config .Export
53
53
54
- config_path = self .get_config_path (self .export_config .weight_path )
54
+ config_path = self .export_config .get ("basic_config_path" , None )
55
+ if not config_path :
56
+ config_path = self .get_config_path (self .export_config .weight_path )
55
57
56
58
self .pdx_config , self .pdx_model = build_model (
57
59
self .global_config .model , config_path = config_path
Original file line number Diff line number Diff line change @@ -50,8 +50,11 @@ def __init__(self, config: AttrDict):
50
50
self .global_config = config .Global
51
51
self .train_config = config .Train
52
52
self .benchmark_config = config .get ("Benchmark" , None )
53
+ config_path = self .train_config .get ("basic_config_path" , None )
53
54
54
- self .pdx_config , self .pdx_model = build_model (self .global_config .model )
55
+ self .pdx_config , self .pdx_model = build_model (
56
+ self .global_config .model , config_path = config_path
57
+ )
55
58
56
59
def train (self , * args , ** kwargs ):
57
60
"""execute model training"""
You can’t perform that action at this time.
0 commit comments