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

SSDLoss implementation

Browse source
This commit is contained in:
Corentin 2021-05-21 15:14:14 +09:00
commit d87bb89e6c
2 changed files with 113 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
layers.py
View file

@ -54,7 +54,7 @@ class Linear(Layer):
def __init__(self, in_channels: int, out_channels: int, activation=0, batch_norm=None, **kwargs):
super().__init__(activation, batch_norm)
self.fc = nn.Linear(in_channels, out_channels, **kwargs)
self.fc = nn.Linear(in_channels, out_channels, bias=not self.batch_norm, **kwargs)
self.batch_norm = nn.BatchNorm1d(
out_channels,
momentum=Layer.BATCH_NORM_MOMENTUM,