fix bug

open-mmlab · Jan 15, 2023 · ffed2c9 · ffed2c9
1 parent aceee13
commit ffed2c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmedit/models/losses/gan_loss.py b/mmedit/models/losses/gan_loss.py
@@ -421,7 +421,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
 def r1_gradient_penalty_loss(discriminator: nn.Module,
                              real_data: torch.Tensor,
-                             mask: Optional(torch.Tensor) = None,
+                             mask: Optional[torch.Tensor] = None,
                              norm_mode: str = 'pixel',
                              loss_scaler: Optional[GradScaler] = None,
                              use_apex_amp: bool = False) -> torch.Tensor: