Tikquuss/sag_torch

Minimizing Finite Sums with Stochastic Average Gradient (SAG)

git clone https://github.com/Tikquuss/sag_torch

cd sag_torch

pip install -r requirements.txt

chmod +x train.sh 
./train.sh 

# or (to loop over hyperparameters)

chmod +x train_loop.sh
./train_loop.sh 

See the notebook Train.ipynb