(self, inchan, nblock, outchan, scale=1, pool=True, **kwargs)
| 77 | """ |
| 78 | |
| 79 | def __init__(self, inchan, nblock, outchan, scale=1, pool=True, **kwargs): |
| 80 | super().__init__() |
| 81 | self.inchan = inchan |
| 82 | self.outchan = outchan |
| 83 | self.pool = pool |
| 84 | self.firstconv = NormedConv2d(inchan, outchan, 3, padding=1) |
| 85 | s = scale / math.sqrt(nblock) |
| 86 | self.blocks = nn.ModuleList( |
| 87 | [CnnBasicBlock(outchan, scale=s, **kwargs) for _ in range(nblock)] |
| 88 | ) |
| 89 | |
| 90 | def forward(self, x): |
| 91 | x = self.firstconv(x) |
no test coverage detected