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 torch.optim as optim
import itertools import itertools
from model.warplayer import warp from model.warplayer import warp
from torchstat import stat
from torch.nn.parallel import DistributedDataParallel as DDP from torch.nn.parallel import DistributedDataParallel as DDP
from model.IFNet import * from model.IFNet import *
import torch.nn.functional as F import torch.nn.functional as F