From 26378496972d1922abefaf585a179fa7a4a84d85 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Wed, 21 Apr 2021 12:03:28 +0800
Subject: [PATCH] Update loss.py

---
 model/loss.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/model/loss.py b/model/loss.py
index e7454e9..72e5de6 100644
--- a/model/loss.py
+++ b/model/loss.py
@@ -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")