mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-16 16:37:51 +01:00
Update RIFE.py
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user