Move HD model

This commit is contained in:
hzwer
2021-09-02 17:38:01 +08:00
parent ea632dd3fb
commit 335bff589a
5 changed files with 5 additions and 5 deletions

View File

@@ -6,7 +6,7 @@ import torch.optim as optim
import itertools
from model.warplayer import warp
from torch.nn.parallel import DistributedDataParallel as DDP
from model.IFNet_HD import *
from model.oldmodel.IFNet_HD import *
import torch.nn.functional as F
from model.loss import *

View File

@@ -6,7 +6,7 @@ import torch.optim as optim
import itertools
from model.warplayer import warp
from torch.nn.parallel import DistributedDataParallel as DDP
from model.IFNet_HDv2 import *
from model.oldmodel.IFNet_HDv2 import *
import torch.nn.functional as F
from model.loss import *