Hi,
I strongly suggest you to use normalization. The layers don’t have normalized data being fed to them and this as such has slowed down their training–try using BatchNorm1d(channels) in between the linear layers.