corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Residual blocks, precache for BatchGenerator

Browse source
This commit is contained in:
Corentin Risselin 2020-09-09 02:27:59 +09:00
commit 5081cf63fe
3 changed files with 139 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
layers.py
View file

@ -50,7 +50,8 @@ class Conv2d(Layer):
stride: Union[int, Tuple[int, int]] = 1, activation=0, batch_norm=None, **kwargs):
super().__init__(activation, batch_norm)
self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=kernel_size, stride=stride, **kwargs)
self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=kernel_size, stride=stride,
bias=not self.batch_norm, **kwargs)
self.batch_norm = nn.BatchNorm2d(
out_channels,
momentum=Layer.BATCH_NORM_MOMENTUM,