2022-11-07 09:36:41 +00:00
|
|
|
import lightning as L
|
|
|
|
|
from lightning.pytorch.demos.boring_classes import BoringModel
|
2022-07-29 14:44:52 +00:00
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
model = BoringModel()
|
2022-11-07 09:36:41 +00:00
|
|
|
trainer = L.Trainer(max_epochs=1)
|
2022-07-29 14:44:52 +00:00
|
|
|
trainer.fit(model)
|
