Neural Networks with PyTorch and Lightning AI Part 5: Final Results and GPU Acceleration (opens in new tab)

In the , we saw how we automated several manual pieces when training a neural network. In this article, we will check the final results and also see how we can do hardware acceleration. For verifying that the final_bias is correctly optimized, we can print out the new value: trainer.fit(model, train_dataloaders=dataloader) print(model.final_bias.data) This gives us -16.0098. Plotting the Result Let’s also visualize it: output_values = model(input_doses) sns.set(style="whitegrid") sns.lineplot...