From 96ccdb5c903e85a60fe51b8735c9a58c4a3ff44d Mon Sep 17 00:00:00 2001
From: Brian Bjarke Jensen <bbj@skov.dk>
Date: Sun, 28 Jul 2024 21:16:32 +0200
Subject: [PATCH] moved file for clarity

---
 model/src/{ => utils}/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
 rename model/src/{ => utils}/train.py (95%)

diff --git a/model/src/train.py b/model/src/utils/train.py
similarity index 95%
rename from model/src/train.py
rename to model/src/utils/train.py
index 20c7370..4d84f24 100644
--- a/model/src/train.py
+++ b/model/src/utils/train.py
@@ -6,7 +6,7 @@ import torch
 import torch.nn as nn
 from dataloader import VCDADataset  # noqa: F401
 
-from .models import VisualCommunicationModel
+from ..models import VisualCommunicationModel
 
 PRE_WARMUP_LR = 1e-10
 POST_WARMUP_LR = 1e-5