Some minor changes.

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."""