Hi, I think the autoencoder architecture is not proper to do a regression.
If you are not familiar with this, I recommend you to look more about classification architectures such as LeNet, VGG, ResNet, …
It is possible to modify the last layer nn.Linear(xxxx, num_classes) to nn.Linear(xxxx, 3)
and use somewhat regression loss functions.