diff options
author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-18 20:56:19 +0100 |
---|---|---|
committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-11-18 20:56:19 +0100 |
commit | 527bb98b191d82b308de1585047e06056258d08d (patch) | |
tree | f33145dba398825871da3184a2735f6fb0b07268 /src/training/run_experiment.py | |
parent | f2cd16f340aa11afadb8fa90c29f85ca1b75a600 (diff) |
Some minor changes.
Diffstat (limited to 'src/training/run_experiment.py')
-rw-r--r-- | src/training/run_experiment.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/training/run_experiment.py b/src/training/run_experiment.py index 55a9572..a883b45 100644 --- a/src/training/run_experiment.py +++ b/src/training/run_experiment.py @@ -21,8 +21,9 @@ from training.trainer.train import Trainer import wandb import yaml - +import text_recognizer.models from text_recognizer.models import Model +import text_recognizer.networks from text_recognizer.networks.loss import loss as custom_loss_module EXPERIMENTS_DIRNAME = Path(__file__).parents[0].resolve() / "experiments" @@ -77,13 +78,12 @@ def _load_modules_and_arguments(experiment_config: Dict,) -> Tuple[Callable, Dic dataset_ = dataset_args["type"] # Import the model module and model arguments. - models_module = importlib.import_module("text_recognizer.models") - model_class_ = getattr(models_module, experiment_config["model"]) + model_class_ = getattr(text_recognizer.models, experiment_config["model"]) # Import metrics. metric_fns_ = ( { - metric: getattr(models_module, metric) + metric: getattr(text_recognizer.networks, metric) for metric in experiment_config["metrics"] } if experiment_config["metrics"] is not None |