glenn-jocher
commited on
Commit
•
ba48f86
1
Parent(s):
f010147
Add bias to Classify() (#1601)
Browse files- 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
|
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):
|