1 #pragma once 2 3 #include <torch/optim/adagrad.h> 4 #include <torch/optim/adam.h> 5 #include <torch/optim/adamw.h> 6 #include <torch/optim/lbfgs.h> 7 #include <torch/optim/optimizer.h> 8 #include <torch/optim/rmsprop.h> 9 #include <torch/optim/sgd.h> 10 11 #include <torch/optim/schedulers/lr_scheduler.h> 12 #include <torch/optim/schedulers/reduce_on_plateau_scheduler.h> 13 #include <torch/optim/schedulers/step_lr.h> 14