feed-forward neural networks #train network using rmsprop optimizer rmsprp_optimizer tch optim rmsprop(model parameters(lr= alpha= eps= - weight_decay= momentum= centered=trueprint("rmsprop "rmsprop_loss train_network(model,rmsprp_optimizer,loss_ function ,num_epochs,batch_size,x_train,y_train#train network using adam optimizer model neuralnetwork(adam_optimizer tch optim adam(model parameters(),lr print("adam "adam_loss train_network(model,adam_optimizer,loss_function ,num_epochs,batch_size,x_train,y_train#train network using sgd optimizer model neuralnetwork(sgd_optimizer tch optim sgd(model parameters()lr= momentum= print("sgd "sgd_loss train_network(model,sgd_optimizer,loss_function ,num_epochs,batch_size,x_train,y_train#plot the losses for each optimizer across epochs import matplotlib pyplot as plt %matplotlib inline epochs range( ,