Fix Generator

This commit is contained in:
henryruhs
2025-03-11 14:43:09 +01:00
parent dc0abff0ce
commit 206a1411d1
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -13,7 +13,7 @@ CONFIG.read('config.ini')
class Generator(nn.Module):
def __init__(self) -> None:
super(Generator, self).__init__()
encoder_type = CONFIG.getint('training.model.generator', 'encoder_type')
encoder_type = CONFIG.get('training.model.generator', 'encoder_type')
id_channels = CONFIG.getint('training.model.generator', 'id_channels')
num_blocks = CONFIG.getint('training.model.generator', 'num_blocks')