Danny Wells

Hi Carlo, Unfortunately, the error is still there. I think I have narrowed it down: ``` >>> scpoli_model.train( ... n_epochs=50, ... pretraining_epochs=51, ... early_stopping_kwargs=early_stopping_kwargs, ... eta=5, ... ) |████████████████████| 100.0%...

@cdedonno - the traceback does not, but as I mentioned above, I think it is happening at line 370 of scpoli/trainer.py My sense is, as you are transitioning from pretraining...

Hi @cdedonno here is the entirety of the code - it is from the tutorial on scpoli: ``` import os import torch import numpy as np import scanpy as sc...

No worries. Really appreciate all the help.