moved file for clarity
This commit is contained in:
@@ -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
|
||||||
Reference in New Issue
Block a user