Removing "water decoder" references

omnigan/generator.py

@@ -81,9 +81,6 @@ def __init__(self, opts, latent_shape=None, verbose=None):
         if "s" in opts.tasks and not opts.gen.s.ignore:
             self.decoders["s"] = SegmentationDecoder(opts)
 
-        if "w" in opts.tasks and not opts.gen.w.ignore:
-            self.decoders["w"] = WaterDecoder(opts)
-
         if "m" in opts.tasks and not opts.gen.m.ignore:
             self.decoders["m"] = MaskDecoder(opts)
 
@@ -225,19 +222,6 @@ def __init__(self, opts):
         )
 
 
-class WaterDecoder(BaseDecoder):
-    def __init__(self, opts):
-        super().__init__(
-            opts.gen.w.n_upsample,
-            opts.gen.w.n_res,
-            opts.gen.w.res_dim,
-            opts.gen.w.output_dim,
-            res_norm=opts.gen.w.res_norm,
-            activ=opts.gen.w.activ,
-            pad_type=opts.gen.w.pad_type,
-        )
-
-
 class MaskDecoder(BaseDecoder):
     def __init__(self, opts):
         super().__init__(

shared/trainer/defaults.yaml

@@ -2,7 +2,7 @@ output_path: /network/tmp1/schmidtv/yb_runs/test_v1
 # -------------------
 # -----  Tasks  -----
 # -------------------
-tasks: [a, d, h, s, t, w, m]
+tasks: [a, d, h, s, t, m]
 
 # ----------------
 # ----- Data -----