Regression Estimator

Loss

Compare

Usage

Requirements

• Python3

Training

• $ python train.py --valid --verbose
• the evaluation result will be saved at ./evaluations
• the model's weight will be saved as ./weights/estimator.ckpt

Function Call in C

• convert PyTorch model to Torch script $ python convert.py
• the Torch script module will be saved as ./weights/traced_regression_net_model.pt
• donwload libtorch zip file and unzip it
• move libtorch to /usr/local/lib
• compile estimator
  • $ cd lib/estimator/
  • $ mkdir build
  • $ cd build
  • $ cmake ..
  • $ cmake --build . --config Release
  • $ mkdir libestimator
  • $ mv libestimator.so libestimator
• move libestimator to /usr/local/lib
• $ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/libestimator
• $ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/libtorch/lib
• $ make
• $ ./pwm_estimator