INNER CODE UNIT · Python
__init__
RichardS0268/CNN-for-Trading · model.py:15
def __init__(self):
super(CNN5d, self).__init__()
self.conv1 = nn.Sequential(OrderedDict([
('Conv', nn.Conv2d(1, 64, (5, 3), padding=(2, 1), stride=(1, 1), dilation=(1, 1))), # output size: [N, 64, 32, 15]
('BN', nn.BatchNorm2d(64, affine=True)),
('ReLU', nn.ReLU()),
('Max-Pool', nn.MaxPool2d((2,1))) # output size: [N, 64, 16, 15]
]))
self.conv1 = self.conv1.apply(self.init_weights)
self.conv2 = nn.Sequential(OrderedDict([
('Conv', nn.Conv2d(64, 128, (5, 3), padding=(2, 1), stride=(1, 1), dilation=(1, 1))), # output size: [N, 128, 16, 15]
('BN', nn.BatchNorm2d(128, affine=True)),
('ReLU', nn.ReLU()),
('Max-Pool', nn.MaxPool2d((2,1))) # output size: [N, 128, 8, 15]
]))
self.conv2 = self.conv2.apply(self.init_weights)