Loading saved models for inference can be heavily depending on the model type. In particular, only the generator-model must be loaded for (Wasserstein) GAN models for inference. The current approach via an if-statement in the get_model_info-method in postprocess.py is farily hacky (cf. here and ungeneric.
In this branch, a load_inference_model is therefore added to the AbstractModelClass which can be overwritten in custom models if required. Additionally, the get_model_info-method will be revised to directly load the checkpointed model for inference.
Note that the corresponding issue-branch is based on #37. Thus, it is suifficient to only merge this branch into develop after accomplishing work in both isssues.
Loading saved models for inference can be heavily depending on the model type. In particular, only the generator-model must be loaded for (Wasserstein) GAN models for inference. The current approach via an if-statement in the
get_model_info
-method inpostprocess.py
is farily hacky (cf. here and ungeneric. In this branch, aload_inference_model
is therefore added to theAbstractModelClass
which can be overwritten in custom models if required. Additionally, theget_model_info
-method will be revised to directly load the checkpointed model for inference.Note that the corresponding issue-branch is based on #37. Thus, it is suifficient to only merge this branch into
develop
after accomplishing work in both isssues.