diff options
Diffstat (limited to 'src/training/trainer')
| -rw-r--r-- | src/training/trainer/train.py | 4 |
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.""" |