This commit is contained in:
henryruhs
2025-03-11 19:43:24 +01:00
parent 5550c89f43
commit ecc5b1a1d5
+1 -1
View File
@@ -15,7 +15,7 @@ def test_aad_with_unet(output_size : int) -> None:
{
'training.model.generator':
{
'identity_channels': '512',
'source_channels': '512',
'output_channels': str(output_size * 16),
'output_size': str(output_size),
'num_blocks': '2'