{'dilations': [1, 1], 'group': 1, 'kernel_shape': [5, 5], 'pads': [2, 2, 2, 2], 'strides': [2, 2]}
the pytorch params show 'strides', (with a s at the end)
code:
if "stride" in self.params: stride = self.params["stride"]
has no 's' at the end, so the graph title will not have the stride info
{'dilations': [1, 1], 'group': 1, 'kernel_shape': [5, 5], 'pads': [2, 2, 2, 2], 'strides': [2, 2]} the pytorch params show 'strides', (with a s at the end)
code:
if "stride" in self.params: stride = self.params["stride"]has no 's' at the end, so the graph title will not have the stride info