Remove the comma after nn.Flatten()
and the error should be gone:
self.flatten = nn.Flatten (), # <- note the comma
Remove the comma after nn.Flatten()
and the error should be gone:
self.flatten = nn.Flatten (), # <- note the comma