The following changes can work!
acc_vals_validation = acc_vals_validation[:(final_epoch + 1)]
loss_vals_training = loss_vals_training[:(final_epoch + 1)]
loss_vals_validation = loss_vals_validation[:(final_epoch + 1)]
loss_std_validation = loss_std_validation[:(final_epoch + 1)]
loss_std_training = loss_std_training[:(final_epoch + 1)]
I have checked this code .