gitextract_rhnta46k/

├── CONTRIBUTING.md
├── LICENSE
├── README.md
├── rigl/
│   ├── __init__.py
│   ├── cifar_resnet/
│   │   ├── data_helper.py
│   │   ├── data_helper_test.py
│   │   ├── resnet_model.py
│   │   └── resnet_train_eval.py
│   ├── experimental/
│   │   └── jax/
│   │       ├── README.md
│   │       ├── __init__.py
│   │       ├── analysis/
│   │       │   └── plot_summary_json.ipynb
│   │       ├── datasets/
│   │       │   ├── __init__.py
│   │       │   ├── cifar10.py
│   │       │   ├── cifar10_test.py
│   │       │   ├── dataset_base.py
│   │       │   ├── dataset_base_test.py
│   │       │   ├── dataset_factory.py
│   │       │   ├── dataset_factory_test.py
│   │       │   ├── mnist.py
│   │       │   └── mnist_test.py
│   │       ├── fixed_param.py
│   │       ├── fixed_param_test.py
│   │       ├── models/
│   │       │   ├── __init__.py
│   │       │   ├── cifar10_cnn.py
│   │       │   ├── cifar10_cnn_test.py
│   │       │   ├── mnist_cnn.py
│   │       │   ├── mnist_cnn_test.py
│   │       │   ├── mnist_fc.py
│   │       │   ├── mnist_fc_test.py
│   │       │   ├── model_factory.py
│   │       │   └── model_factory_test.py
│   │       ├── prune.py
│   │       ├── prune_test.py
│   │       ├── pruning/
│   │       │   ├── __init__.py
│   │       │   ├── init.py
│   │       │   ├── init_test.py
│   │       │   ├── mask_factory.py
│   │       │   ├── mask_factory_test.py
│   │       │   ├── masked.py
│   │       │   ├── masked_test.py
│   │       │   ├── pruning.py
│   │       │   ├── pruning_test.py
│   │       │   ├── symmetry.py
│   │       │   └── symmetry_test.py
│   │       ├── random_mask.py
│   │       ├── random_mask_test.py
│   │       ├── requirements.txt
│   │       ├── run.sh
│   │       ├── shuffled_mask.py
│   │       ├── shuffled_mask_test.py
│   │       ├── train.py
│   │       ├── train_test.py
│   │       ├── training/
│   │       │   ├── __init__.py
│   │       │   ├── training.py
│   │       │   └── training_test.py
│   │       └── utils/
│   │           ├── __init__.py
│   │           ├── utils.py
│   │           └── utils_test.py
│   ├── imagenet_resnet/
│   │   ├── colabs/
│   │   │   ├── MobileNet_Counting.ipynb
│   │   │   └── Resnet_50_Param_Flops_Counting.ipynb
│   │   ├── imagenet_train_eval.py
│   │   ├── mobilenetv1_model.py
│   │   ├── mobilenetv2_model.py
│   │   ├── pruning_layers.py
│   │   ├── resnet_model.py
│   │   ├── train_test.py
│   │   ├── utils.py
│   │   └── vgg.py
│   ├── mnist/
│   │   ├── mnist_train_eval.py
│   │   └── visualize_mask_records.py
│   ├── requirements.txt
│   ├── rigl_tf2/
│   │   ├── README.md
│   │   ├── colabs/
│   │   │   └── MnistProp.ipynb
│   │   ├── configs/
│   │   │   ├── dense.gin
│   │   │   ├── grasp.gin
│   │   │   ├── hessian.gin
│   │   │   ├── interpolate.gin
│   │   │   ├── lottery.gin
│   │   │   ├── prune.gin
│   │   │   ├── rigl.gin
│   │   │   ├── scratch.gin
│   │   │   ├── set.gin
│   │   │   ├── small_dense.gin
│   │   │   └── snip.gin
│   │   ├── init_utils.py
│   │   ├── interpolate.py
│   │   ├── mask_updaters.py
│   │   ├── metainit.py
│   │   ├── mlp_configs/
│   │   │   ├── dense.gin
│   │   │   ├── lottery.gin
│   │   │   ├── prune.gin
│   │   │   ├── rigl.gin
│   │   │   ├── scratch.gin
│   │   │   ├── set.gin
│   │   │   └── small_dense.gin
│   │   ├── networks.py
│   │   ├── train.py
│   │   └── utils.py
│   ├── rl/
│   │   ├── README.md
│   │   ├── dqn_agents.py
│   │   ├── requirements.txt
│   │   ├── run.sh
│   │   ├── run_experiment.py
│   │   ├── sparse_utils.py
│   │   ├── sparsetrain_configs/
│   │   │   ├── dqn_atari_dense.gin
│   │   │   ├── dqn_atari_dense_impala_net.gin
│   │   │   ├── dqn_atari_prune.gin
│   │   │   ├── dqn_atari_prune_impala_net.gin
│   │   │   ├── dqn_atari_rigl.gin
│   │   │   ├── dqn_atari_rigl_impala_net.gin
│   │   │   ├── dqn_atari_set.gin
│   │   │   ├── dqn_atari_set_impala_net.gin
│   │   │   ├── dqn_atari_static.gin
│   │   │   └── dqn_atari_static_impala_net.gin
│   │   ├── tfagents/
│   │   │   ├── configs/
│   │   │   │   ├── dqn_gym_dense_config.gin
│   │   │   │   ├── dqn_gym_pruning_config.gin
│   │   │   │   ├── dqn_gym_sparse_config.gin
│   │   │   │   ├── ppo_mujoco_dense_config.gin
│   │   │   │   ├── ppo_mujoco_pruning_config.gin
│   │   │   │   ├── ppo_mujoco_sparse_config.gin
│   │   │   │   ├── sac_mujoco_dense_config.gin
│   │   │   │   ├── sac_mujoco_pruning_config.gin
│   │   │   │   └── sac_mujoco_sparse_config.gin
│   │   │   ├── dqn_train_eval.py
│   │   │   ├── ppo_train_eval.py
│   │   │   ├── sac_train_eval.py
│   │   │   ├── sparse_encoding_network.py
│   │   │   ├── sparse_ppo_actor_network.py
│   │   │   ├── sparse_ppo_discrete_actor_network.py
│   │   │   ├── sparse_ppo_discrete_actor_network_test.py
│   │   │   ├── sparse_tanh_normal_projection_network.py
│   │   │   ├── sparse_value_network.py
│   │   │   └── tf_sparse_utils.py
│   │   └── train.py
│   ├── sparse_optimizers.py
│   ├── sparse_optimizers_base.py
│   ├── sparse_optimizers_test.py
│   ├── sparse_utils.py
│   ├── sparse_utils_test.py
│   └── str_sparsities.py
└── run.sh