image_through_model #49

Merged
brian merged 75 commits from image_through_model into main 2024-10-20 00:10:27 +02:00
Showing only changes of commit 96ccdb5c90 - Show all commits
@@ -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