Skip to content

Latest commit

 

History

History
35 lines (25 loc) · 590 Bytes

README.md

File metadata and controls

35 lines (25 loc) · 590 Bytes

Torch-Wrapper

A wrapper for calling scipy.optimize.minimize on torch Module

wiki/test-functions-for-optimization

install

pip install git+https://github.com/husisy/torch-wrapper.git

# or you may git clone the repo and install it locally
git clone git+https://github.com/husisy/torch-wrapper.git
pip install .

# for developer
pip install -e .

quickstart

cd example
python draft00.py
# Rastrigin function
# Ackley function
# Rosenbrock function
# Beale function

unittest

pytest