Closed adizhol-gp closed 7 months ago
update: seems like changing the pytorch_lightning import to "import lightning.pytorch" solves the problem
I'll merge PR #370 soon enough to move from the old pytorch_lightning interface to lightning.
Merged it, everything should be fine.
using SwaV: when calling trainer.fit(model):
error:
model must be a LightningModule
versions: