Update RIFE15C.py

This commit is contained in:
hzwer
2021-04-26 15:18:55 +08:00
committed by GitHub
parent adc1957ef8
commit 6ff1745847

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 import *
from model.IFNet15C import *
import torch.nn.functional as F
from model.loss import *