Skip to content

SMRhadou/RobustUnrolling

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RobustUnrolling

To train constrained LISTA (or use the RunMe.sh file)

python3 LISTA.py --nLayers 10 --lr 1e-5 --lr_dual 1e-3 --eps 0.05 --noisyOuts --constrained --Trial exp_name

To train standard LISTA

python3 LISTA.py --nLayers 10 --lr 1e-6 --eps 0.05 --Trial exp_name

To train standard LISTA with noise

python3 LISTA.py --nLayers 10 --lr 1e-6 --eps 0.05 --noisyOuts --Trial exp_name

For testing and generating the figures

python3 testing.py 

About

Implementation of LISTA with descending constraints as in the paper "Robust Stochastically-Descending Unrolled Netowrks"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors