image_through_model #49
@@ -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
|
||||
Reference in New Issue
Block a user