glenn-jocher commited on
Commit
ba48f86
1 Parent(s): f010147

Add bias to Classify() (#1601)

Browse files
Files changed (1) hide show
  1. models/common.py +1 -2
models/common.py CHANGED
@@ -1,7 +1,6 @@
1
  # This file contains modules common to various models
2
 
3
  import math
4
-
5
  import numpy as np
6
  import torch
7
  import torch.nn as nn
@@ -244,7 +243,7 @@ class Classify(nn.Module):
244
  def __init__(self, c1, c2, k=1, s=1, p=None, g=1): # ch_in, ch_out, kernel, stride, padding, groups
245
  super(Classify, self).__init__()
246
  self.aap = nn.AdaptiveAvgPool2d(1) # to x(b,c1,1,1)
247
- self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False) # to x(b,c2,1,1)
248
  self.flat = Flatten()
249
 
250
  def forward(self, x):
 
1
  # This file contains modules common to various models
2
 
3
  import math
 
4
  import numpy as np
5
  import torch
6
  import torch.nn as nn
 
243
  def __init__(self, c1, c2, k=1, s=1, p=None, g=1): # ch_in, ch_out, kernel, stride, padding, groups
244
  super(Classify, self).__init__()
245
  self.aap = nn.AdaptiveAvgPool2d(1) # to x(b,c1,1,1)
246
+ self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g) # to x(b,c2,1,1)
247
  self.flat = Flatten()
248
 
249
  def forward(self, x):