The code is tested with Python 3.10
pip install -r requirements.txt
git clone [email protected]:zhou-yh19/PyDeePC.git
cd PyDeePC
pip install -e .
src/modules
: PyTorch modules for approximating the scoring functionsrc/scripts
: Scripts for data augmentation, and computing different scoring functionsrc/utils
: Utility functions (customized PyTorch operations, tensor operations, etc.)src/train.py
: Training script for the scoring functionsrc/grid_train_one_gpu.py
: Training script for the scoring function on a single GPUexperiments
: Sample scripts for running experimentstest
: julia scripts for testing solver
The project is released under the MIT license. See LICENSE for details.
Part of the project is modified from PyDeePC