def get_model(input_shape, nclasses=3): if starting_model_fp is None: return ku.model_c3(input_shape, nclasses, compile_model=True) else: return load_model(starting_model_fp)
def get_model(input_shape, nclasses=3): if starting_model_fp is None and trn_passes != 0: # not an eval only script return ku.model_c3(input_shape, nclasses, compile_model=True) else: return load_model(starting_model_fp)