Searched defs:AdamWOptions (Results 1 – 2 of 2) sorted by relevance
20 struct TORCH_API AdamWOptions : public OptimizerCloneableOptions<AdamWOptions> { struct33 const AdamWOptions& lhs, argument
17 AdamWOptions::AdamWOptions(double lr) : lr_(lr) {} in AdamWOptions() function in torch::optim::AdamWOptions