Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-09-01 19:50:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update loss.py

Browse Source
This commit is contained in:
hzwer
2021-04-21 12:03:28 +08:00
committed by GitHub
parent 1dc514b650
commit 2637849697
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
model/loss.py
View File

@@ -2,6 +2,7 @@ import torch
import numpy as np
import torch.nn as nn
import torch.nn.functional as F
import torchvision.models as models
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")