Update RIFE.py

This commit is contained in:
hzwer
2021-10-18 13:55:10 +08:00
committed by GitHub
parent 22f6372b8a
commit eeee0df2c6

View File

@@ -5,7 +5,6 @@ from torch.optim import AdamW
import torch.optim as optim
import itertools
from model.warplayer import warp
from torchstat import stat
from torch.nn.parallel import DistributedDataParallel as DDP
from model.IFNet import *
import torch.nn.functional as F