From 527bb98b191d82b308de1585047e06056258d08d Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Wed, 18 Nov 2020 20:56:19 +0100
Subject: Some minor changes.

---
 src/training/trainer/train.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/training/trainer/train.py')

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."""
 
-- 
cgit v1.2.3-70-g09d2