def weights(self):
_weights = {
'W_conv1': ne.weight_variable([5, 5, 1, 32], name='W_conv1'),
'W_conv2': ne.weight_variable([5, 5, 32, 64], name='W_conv2'),
'W_fc1': ne.weight_variable([7 * 7 * 64, 1024], name='W_fc1'),
'W_fc2': ne.weight_variable([1024, 10], name='W_fc2')
}
return _weights
def weights(self):
_weights = {
'W_conv1': ne.weight_variable([3, 9, 1, 16], name='W_conv1'),
'W_conv2': ne.weight_variable([2, 8, 16, 32], name='W_conv2'),
'W_conv3': ne.weight_variable([3, 6, 32, 64], name='W_conv3'),
'W_fc1': ne.weight_variable([1 * 6 * 64, 128], name='W_fc1'),
'W_fc2': ne.weight_variable([128, self.class_num], name='W_fc2')
}
return _weights
def weights(self):
_weights = {
'W_conv1': ne.weight_variable([3, 3, 1, 48], name='W_conv1'),
'W_conv2': ne.weight_variable([3, 3, 48, 48], name='W_conv2'),
'W_conv3': ne.weight_variable([3, 3, 48, 48], name='W_conv3'),
'W_conv4': ne.weight_variable([3, 3, 48, 48], name='W_conv4'),
'W_conv5': ne.weight_variable([3, 3, 48, 48], name='W_conv5'),
'W_conv6': ne.weight_variable([3, 3, 48, 48], name='W_conv6'),
'W_conv7': ne.weight_variable([3, 3, 48, 48], name='W_conv7'),
'W_conv8': ne.weight_variable([1, 1, 48, 48], name='W_conv8'),
'W_conv9': ne.weight_variable([1, 1, 48, 1], name='W_conv9')
}
return _weights
