Remove the last F.relu so that your model is able to return negative and positive logits and rerun the script.
If that doesn’t help, try to overfit a small dataset (e.g. just 10 samples) by playing around with the hyperparameters.