text: /**
   * Copyright (c) Facebook, Inc. and its affiliates.
   * All rights reserved.
   *
   * This source code is licensed under the BSD-style license found in the
   * LICENSE file in the root directory of this source tree.
   */
  
  #pragma once
  
  #include <flashlight/flashlight.h>
  
  #include "common/Defines.h"
  
  namespace w2l {
  
  std::shared_ptr<fl::FirstOrderOptimizer> initOptimizer(
      const std::vector<std::shared_ptr<fl::Module>>& nets,
      const std::string& optimizer,
      double lr,
      double momentum,
      double weightdecay);
  } // namespace w2l
  