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