fix

Linaom1214 · May 16, 2022 · 1326c7b · 1326c7b
1 parent 6fb62f8
commit 1326c7b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -4,7 +4,6 @@
 import os
 from datetime import datetime
 from copy import deepcopy
-from torchmetrics import AUC
 from tqdm import tqdm
 from skimage.segmentation import mark_boundaries
 import matplotlib.pyplot as plt
@@ -20,7 +19,6 @@
 from torchvision import transforms
 from torchvision.transforms.functional import to_pil_image
 from torchvision.utils import save_image
-from zmq import device
 
 from net.Unet import Unet
 from net.FCN import fcn