summaryrefslogtreecommitdiff
path: root/src/training/trainer
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/trainer')
-rw-r--r--src/training/trainer/train.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/training/trainer/train.py b/src/training/trainer/train.py
index 223d9c6..8ae994a 100644
--- a/src/training/trainer/train.py
+++ b/src/training/trainer/train.py
@@ -3,6 +3,7 @@
from pathlib import Path
import time
from typing import Dict, List, Optional, Tuple, Type
+import warnings
from einops import rearrange
from loguru import logger
@@ -23,6 +24,9 @@ torch.manual_seed(4711)
torch.cuda.manual_seed(4711)
+warnings.filterwarnings("ignore")
+
+
class Trainer:
"""Trainer for training PyTorch models."""