This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent 8f1f63f2ef
commit d944d95bfd
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -5,7 +5,7 @@ from typing import Tuple, cast
import torch
import torchvision
from lightning import Trainer, LightningModule
from lightning import LightningModule, Trainer
from lightning.pytorch.callbacks import ModelCheckpoint
from lightning.pytorch.loggers import TensorBoardLogger
from torch import Tensor, nn