moved file for clarity

This commit is contained in:
Brian Bjarke Jensen
2024-07-28 21:16:32 +02:00
parent 6e9c0a2626
commit 96ccdb5c90
@@ -6,7 +6,7 @@ import torch
import torch.nn as nn import torch.nn as nn
from dataloader import VCDADataset # noqa: F401 from dataloader import VCDADataset # noqa: F401
from .models import VisualCommunicationModel from ..models import VisualCommunicationModel
PRE_WARMUP_LR = 1e-10 PRE_WARMUP_LR = 1e-10
POST_WARMUP_LR = 1e-5 POST_WARMUP_LR = 1e-5