Spaces:

csgaobb
/

MetaUAS

Running on Zero

csgaobb commited on Mar 27

Commit

c4c425b

1 Parent(s): 295f637

fix a bug in metauas

Files changed (1) hide show

metauas.py CHANGED Viewed

@@ -179,7 +179,7 @@ class MetaUAS(pl.LightningModule):
                 decoder_channels=decoder_channels,
                 n_blocks= self.decoder_depth,
                 attention_type="scse",
-                num_coam_layers= num_alignment_layers,
             )
         elif self.decoder_name == "fpn":

                 decoder_channels=decoder_channels,
                 n_blocks= self.decoder_depth,
                 attention_type="scse",
+                #num_coam_layers= num_alignment_layers,
             )
         elif self.decoder_name == "fpn":