pl_cross

pl_cross copied to clipboard

I noticed that there are also issues with the create_ensemble method complaining that the model argument is missing. I tried it with all possible settings providing a model only, providing checkpoints only, and providing a model and checkpoints. This issue occurs with PyTorch Lightning 1.5.10. I put together a self-contained example notebook here (https://github.com/rasbt/deeplearning-models/blob/master/pytorch-lightning_ipynb/kfold/kfold-light-cnn-mnist.ipynb) that reproduces the error if you uncomment the create_model code at the bottom.