This commit is contained in:
harisreedhar
2025-03-10 13:49:50 +05:30
committed by henryruhs
parent a3ac4d5ddd
commit 606bf42089
2 changed files with 2 additions and 1 deletions
+1
View File
@@ -14,6 +14,7 @@ split_ratio =
embedder_path =
gazer_path =
motion_extractor_path =
parser_path =
[training.model.generator]
identity_channels =
+1 -1
View File
@@ -46,7 +46,7 @@ class MaskNet(nn.Module):
output_tensor = up_sample(output_tensor)
output_tensor = self.conv(output_tensor)
output_tensor = self.activation(output_tensor)
output_tensor = self.sigmoid(output_tensor)
return output_tensor