fix model mode

samjmolyneux · Dec 17, 2020 · 017fb51 · 017fb51
1 parent 6b8f01d
commit 017fb51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/hovernet/net_desc_fast.py b/models/hovernet/net_desc_fast.py
@@ -299,7 +299,7 @@ class HoVerNet(Net):
     """Initialise HoVer-Net."""
 
     def __init__(self, input_ch=3, nr_types=None, freeze=False):
-        super(NetDesc, self).__init__()
+        super().__init__()
         self.freeze = freeze
         self.nr_types = nr_types
         self.output_ch = 3 if nr_types is None else 4