The method MLRunner.run_inference_for_lightning_models takes a list of checkpoint paths as an argument, but then makes sure that there is only one used (here):
if len(checkpoint_paths) != 1:
raise ValueError(f"This method expects exactly 1 checkpoint for inference, but got {len(checkpoint_paths)}")
We want to change this so that the checkpoints gleaned from a BYOL cross validation run can be used as an ensemble model.
This will use the methods defined in the InnerEyeInference abstract class. We expect that these methods are sufficient, but an extension or redesign may be required.
Use the simple linear regression model as the basis of an exemplar.
It would be great to use another, more realistic model. The UHB Covid-19 model is not a lightning model, how about the Single Cell model?
dumbledad
commented
3 years ago
The method
MLRunner.run_inference_for_lightning_modelstakes a list of checkpoint paths as an argument, but then makes sure that there is only one used (here):We want to change this so that the checkpoints gleaned from a BYOL cross validation run can be used as an ensemble model.
InnerEyeInferenceabstract class. We expect that these methods are sufficient, but an extension or redesign may be required.AB#4219