Add module.train() to move a module back to the autodiff backend#3975
Open
Add module.train() to move a module back to the autodiff backend#3975
module.train() to move a module back to the autodiff backend#3975