summaryrefslogtreecommitdiff
path: root/training/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'training/run.py')
-rw-r--r--training/run.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/training/run.py b/training/run.py
index b8f2178..2089a86 100644
--- a/training/run.py
+++ b/training/run.py
@@ -2,19 +2,19 @@
from typing import Callable, List, Optional, Type
import hydra
+import utils
from loguru import logger as log
from omegaconf import DictConfig
from pytorch_lightning import (
Callback,
LightningDataModule,
LightningModule,
- seed_everything,
Trainer,
+ seed_everything,
)
from pytorch_lightning.loggers import Logger
from torch import nn
from torchinfo import summary
-import utils
def run(config: DictConfig) -> Optional[float]: