├── .core-coveragerc ├── .flake8 ├── .github └── workflows │ ├── cover-ci.yml │ ├── docs.yml │ ├── pre-commit.yml │ ├── stale.yml │ └── test-ci.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── docs ├── 01-get-started │ ├── basic-usage.md │ └── installation.md ├── 02-configuration │ ├── cfg-params.md │ └── configuration.md ├── 03-customization │ ├── custom-environments.md │ ├── custom-models.md │ └── custom-multi-agent-environments.md ├── 04-experiments │ ├── experiment-launcher.md │ └── slurm-details.md ├── 05-monitoring │ ├── custom-metrics.md │ ├── metrics-reference.md │ ├── tensorboard.md │ └── wandb.md ├── 05-troubleshooting │ └── troubleshooting.md ├── 06-architecture │ ├── message-passing.md │ └── overview.md ├── 07-advanced-topics │ ├── action-masking.md │ ├── batched-non-batched.md │ ├── db_sampling.png │ ├── doom_norm.png │ ├── double-buffered.md │ ├── exporting-to-onnx.md │ ├── inactive-agents.md │ ├── multi-policy-training.md │ ├── normalization.png │ ├── normalizations.md │ ├── observer.md │ ├── passing-info.md │ ├── pbt.md │ ├── policy-lag.md │ ├── profiling.md │ ├── serial-mode.md │ └── sync-async.md ├── 08-miscellaneous │ ├── tests.md │ └── v1-to-v2.md ├── 09-environment-integrations │ ├── atari.md │ ├── brax.md │ ├── dmlab.md │ ├── envpool.md │ ├── isaacgym.md │ ├── megaverse.md │ ├── mujoco.md │ ├── nethack.md │ ├── pettingzoo.md │ ├── swarm-rl.md │ └── vizdoom.md ├── 10-huggingface │ └── huggingface.md ├── 11-release-notes │ └── release-notes.md ├── 12-community │ ├── citation.md │ ├── contribution.md │ └── doc-contribution.md ├── CNAME ├── cfg-params.sh └── index.md ├── mkdocs.yml ├── pyproject.toml ├── sample_factory ├── __init__.py ├── algo │ ├── __init__.py │ ├── evaluators │ │ ├── __init__.py │ │ └── default_evaluator.py │ ├── learning │ │ ├── __init__.py │ │ ├── batcher.py │ │ ├── learner.py │ │ ├── learner_worker.py │ │ └── rnn_utils.py │ ├── runners │ │ ├── __init__.py │ │ ├── runner.py │ │ ├── runner_parallel.py │ │ └── runner_serial.py │ ├── sampling │ │ ├── __init__.py │ │ ├── batched_sampling.py │ │ ├── evaluation_sampling_api.py │ │ ├── inference_worker.py │ │ ├── non_batched_sampling.py │ │ ├── rollout_worker.py │ │ ├── sampler.py │ │ ├── sampling_utils.py │ │ ├── stats.py │ │ └── sync_sampling_api.py │ └── utils │ │ ├── __init__.py │ │ ├── action_distributions.py │ │ ├── agent_policy_mapping.py │ │ ├── context.py │ │ ├── env_info.py │ │ ├── gymnasium_utils.py │ │ ├── heartbeat.py │ │ ├── make_env.py │ │ ├── misc.py │ │ ├── model_sharing.py │ │ ├── multiprocessing_utils.py │ │ ├── optimizers.py │ │ ├── rl_utils.py │ │ ├── running_mean_std.py │ │ ├── shared_buffers.py │ │ ├── spaces │ │ ├── __init__.py │ │ └── discretized.py │ │ ├── tensor_dict.py │ │ ├── tensor_utils.py │ │ └── torch_utils.py ├── benchmarking │ ├── benchmarking.txt │ ├── benchmarking_isaacgym_ant.txt │ └── benchmarking_xeon_36core.txt ├── cfg │ ├── __init__.py │ ├── arguments.py │ ├── cfg.py │ └── configurable.py ├── enjoy.py ├── envs │ ├── __init__.py │ ├── create_env.py │ ├── env_utils.py │ ├── env_wrappers.py │ └── pettingzoo_envs.py ├── eval.py ├── export_onnx.py ├── huggingface │ ├── __init__.py │ ├── huggingface_utils.py │ ├── load_from_hub.py │ └── push_to_hub.py ├── launcher │ ├── __init__.py │ ├── launcher_utils.py │ ├── run.py │ ├── run_description.py │ ├── run_ngc.py │ ├── run_processes.py │ ├── run_slurm.py │ └── slurm │ │ └── sbatch_timeout.sh ├── model │ ├── __init__.py │ ├── action_parameterization.py │ ├── actor_critic.py │ ├── core.py │ ├── decoder.py │ ├── encoder.py │ ├── model_factory.py │ ├── model_utils.py │ └── utils.py ├── pbt │ ├── __init__.py │ └── population_based_training.py ├── train.py └── utils │ ├── __init__.py │ ├── algo_version.py │ ├── attr_dict.py │ ├── decay.py │ ├── dicts.py │ ├── get_available_gpus.py │ ├── gifs.py │ ├── gpu_utils.py │ ├── network.py │ ├── normalize.py │ ├── tb.py │ ├── timing.py │ ├── typing.py │ ├── utils.py │ └── wandb_utils.py ├── setup.py ├── sf_examples ├── __init__.py ├── atari │ ├── __init__.py │ ├── atari_params.py │ ├── atari_utils.py │ ├── enjoy_atari.py │ ├── experiments │ │ ├── __init__.py │ │ └── atari_envs.py │ └── train_atari.py ├── brax │ ├── __init__.py │ ├── brax_render.py │ ├── enjoy_brax.py │ ├── experiments │ │ ├── __init__.py │ │ ├── brax_basic_envs.py │ │ └── sbatch_timeout_brax.sh │ └── train_brax.py ├── dmlab │ ├── README.md │ ├── __init__.py │ ├── command_line │ │ ├── train_dmlab30.sh │ │ └── train_dmlab30_server_pbt.sh │ ├── dmlab30.py │ ├── dmlab_env.py │ ├── dmlab_gym.py │ ├── dmlab_level_cache.py │ ├── dmlab_model.py │ ├── dmlab_params.py │ ├── dmlab_populate_cache.py │ ├── dmlab_utils.py │ ├── enjoy_dmlab.py │ ├── experiments │ │ ├── __init__.py │ │ └── dmlab30.py │ ├── train_dmlab.py │ └── wrappers │ │ ├── __init__.py │ │ └── reward_shaping.py ├── enjoy_custom_env_custom_model.py ├── enjoy_custom_multi_env.py ├── enjoy_gym_env.py ├── enjoy_pettingzoo_env.py ├── envpool │ ├── __init__.py │ ├── atari │ │ ├── __init__.py │ │ ├── enjoy_envpool_atari.py │ │ ├── envpool_atari_params.py │ │ ├── envpool_atari_utils.py │ │ └── train_envpool_atari.py │ ├── envpool_utils.py │ ├── envpool_wrappers.py │ └── mujoco │ │ ├── __init__.py │ │ ├── enjoy_envpool_mujoco.py │ │ ├── envpool_mujoco_params.py │ │ ├── envpool_mujoco_utils.py │ │ ├── experiments │ │ ├── __init__.py │ │ └── mujoco_envpool.py │ │ └── train_envpool_mujoco.py ├── export_onnx_gym_env.py ├── isaacgym_examples │ ├── __init__.py │ ├── enjoy_isaacgym.py │ ├── experiments │ │ ├── __init__.py │ │ ├── isaacgym_allegrohand.py │ │ ├── isaacgym_ant.py │ │ ├── isaacgym_basic_envs.py │ │ ├── isaacgym_humanoid.py │ │ └── isaacgym_runs.py │ └── train_isaacgym.py ├── mujoco │ ├── README.md │ ├── __init__.py │ ├── enjoy_mujoco.py │ ├── experiments │ │ ├── __init__.py │ │ └── mujoco_all_envs.py │ ├── fast_eval_mujoco.py │ ├── mujoco_params.py │ ├── mujoco_utils.py │ └── train_mujoco.py ├── nethack │ ├── __init__.py │ ├── enjoy_nethack.py │ ├── fast_eval_nethack.py │ ├── models │ │ ├── __init__.py │ │ ├── chaotic_dwarf.py │ │ ├── crop.py │ │ ├── scaled.py │ │ └── utils.py │ ├── nethack_env.py │ ├── nethack_params.py │ ├── scripts │ │ └── sample_env.py │ ├── train_nethack.py │ └── utils │ │ ├── __init__.py │ │ ├── task_rewards.py │ │ └── wrappers │ │ ├── __init__.py │ │ ├── blstats_info.py │ │ ├── no_progress_timeout.py │ │ ├── prev_actions.py │ │ ├── task_rewards.py │ │ └── tile_tty.py ├── notebooks │ └── samplefactory_hub_example.ipynb ├── sampler │ ├── __init__.py │ └── use_simplified_sampling_api.py ├── train_custom_env_custom_model.py ├── train_custom_multi_env.py ├── train_gym_env.py ├── train_pettingzoo_env.py └── vizdoom │ ├── __init__.py │ ├── custom_env │ ├── custom_doom_env.cfg │ └── custom_doom_env.wad │ ├── doom │ ├── README.md │ ├── __init__.py │ ├── action_space.py │ ├── doom_gym.py │ ├── doom_model.py │ ├── doom_params.py │ ├── doom_play_demo.py │ ├── doom_render.py │ ├── doom_utils.py │ ├── multi_agent_match.py │ ├── multiplayer │ │ ├── __init__.py │ │ ├── doom_multiagent.py │ │ └── doom_multiagent_wrapper.py │ ├── play_doom.py │ ├── sample_env.py │ ├── scenarios │ │ ├── basic.cfg │ │ ├── basic.wad │ │ ├── battle.cfg │ │ ├── battle.wad │ │ ├── battle2.wad │ │ ├── battle2_continuous_turning.cfg │ │ ├── battle_continuous_turning.cfg │ │ ├── cig.cfg │ │ ├── cig.wad │ │ ├── deadly_corridor.cfg │ │ ├── deadly_corridor.wad │ │ ├── defend_the_center.cfg │ │ ├── defend_the_center.wad │ │ ├── defend_the_line.cfg │ │ ├── defend_the_line.wad │ │ ├── dwango5.wad │ │ ├── dwango5_dm.cfg │ │ ├── dwango5_dm_continuous_weap.cfg │ │ ├── freedm.cfg │ │ ├── freedm.wad │ │ ├── health_gathering.cfg │ │ ├── health_gathering.wad │ │ ├── health_gathering_supreme.cfg │ │ ├── health_gathering_supreme.wad │ │ ├── my_way_home.cfg │ │ ├── my_way_home.wad │ │ ├── ssl2.cfg │ │ ├── ssl2.wad │ │ ├── two_colors_easy.cfg │ │ ├── two_colors_easy.wad │ │ ├── two_colors_hard.cfg │ │ └── two_colors_hard.wad │ ├── scripts │ │ ├── __init__.py │ │ └── generate_bots.py │ └── wrappers │ │ ├── __init__.py │ │ ├── additional_input.py │ │ ├── bot_difficulty.py │ │ ├── exploration.py │ │ ├── multiplayer_stats.py │ │ ├── observation_space.py │ │ ├── reward_shaping.py │ │ ├── scenario_wrappers │ │ ├── __init__.py │ │ └── gathering_reward_shaping.py │ │ └── step_human_input.py │ ├── enjoy_custom_vizdoom_env.py │ ├── enjoy_vizdoom.py │ ├── experiments │ ├── doom_basic.py │ ├── doom_battle_battle2_appo.py │ ├── doom_battle_d4_appo_pbt.py │ ├── doom_bots_sweep.py │ ├── doom_defend_center.py │ ├── doom_freedm.py │ ├── doom_health_gathering.py │ ├── paper_doom_all_basic_envs.py │ ├── paper_doom_battle2_appo.py │ ├── paper_doom_battle2_appo_pbt.py │ ├── paper_doom_battle_appo.py │ ├── paper_doom_battle_appo_pbt.py │ ├── paper_doom_bots_pbt.py │ ├── paper_doom_duel_bots_pbt.py │ ├── paper_doom_duel_pbt.py │ ├── paper_doom_wall_time.py │ └── sf2_doom_battle_envs.py │ ├── train_custom_vizdoom_env.py │ └── train_vizdoom.py └── tests ├── __init__.py ├── algo ├── __init__.py ├── test_action_distributions.py ├── test_learner.py ├── test_model.py ├── test_model_builder.py ├── test_pbt.py ├── test_rms.py ├── test_rnn.py ├── test_spaces.py └── test_tensordict.py ├── envs ├── __init__.py ├── atari │ ├── __init__.py │ ├── test_atari.py │ └── test_envpool_atari.py ├── dmlab │ ├── __init__.py │ └── test_dmlab.py ├── mujoco │ ├── __init__.py │ ├── test_envpool_mujoco.py │ └── test_mujoco.py ├── pettingzoo │ ├── __init__.py │ └── test_pettingzoo.py ├── tuple_action_envs │ ├── __init__.py │ ├── test_mixed_dist_env_batched.py │ ├── test_mixed_dist_env_non_batched.py │ ├── test_two_discrete_action_dist_env_batched.py │ └── test_two_discrete_action_dist_env_non_batched.py ├── utils.py └── vizdoom │ ├── __init__.py │ ├── test_vizdoom_basic.py │ └── test_vizdoom_multiplayer.py ├── examples ├── __init__.py ├── test_example.py ├── test_example_multi.py └── test_example_sampler.py ├── export_onnx_utils.py ├── test_launcher.py ├── test_precheck.py ├── test_utils.py └── utils.py /.core-coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.core-coveragerc -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/cover-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.github/workflows/cover-ci.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.github/workflows/pre-commit.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.github/workflows/stale.yml -------------------------------------------------------------------------------- /.github/workflows/test-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.github/workflows/test-ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/README.md -------------------------------------------------------------------------------- /docs/01-get-started/basic-usage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/01-get-started/basic-usage.md -------------------------------------------------------------------------------- /docs/01-get-started/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/01-get-started/installation.md -------------------------------------------------------------------------------- /docs/02-configuration/cfg-params.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/02-configuration/cfg-params.md -------------------------------------------------------------------------------- /docs/02-configuration/configuration.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/02-configuration/configuration.md -------------------------------------------------------------------------------- /docs/03-customization/custom-environments.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/03-customization/custom-environments.md -------------------------------------------------------------------------------- /docs/03-customization/custom-models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/03-customization/custom-models.md -------------------------------------------------------------------------------- /docs/03-customization/custom-multi-agent-environments.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/03-customization/custom-multi-agent-environments.md -------------------------------------------------------------------------------- /docs/04-experiments/experiment-launcher.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/04-experiments/experiment-launcher.md -------------------------------------------------------------------------------- /docs/04-experiments/slurm-details.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/04-experiments/slurm-details.md -------------------------------------------------------------------------------- /docs/05-monitoring/custom-metrics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/05-monitoring/custom-metrics.md -------------------------------------------------------------------------------- /docs/05-monitoring/metrics-reference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/05-monitoring/metrics-reference.md -------------------------------------------------------------------------------- /docs/05-monitoring/tensorboard.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/05-monitoring/tensorboard.md -------------------------------------------------------------------------------- /docs/05-monitoring/wandb.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/05-monitoring/wandb.md -------------------------------------------------------------------------------- /docs/05-troubleshooting/troubleshooting.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/05-troubleshooting/troubleshooting.md -------------------------------------------------------------------------------- /docs/06-architecture/message-passing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/06-architecture/message-passing.md -------------------------------------------------------------------------------- /docs/06-architecture/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/06-architecture/overview.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/action-masking.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/action-masking.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/batched-non-batched.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/batched-non-batched.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/db_sampling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/db_sampling.png -------------------------------------------------------------------------------- /docs/07-advanced-topics/doom_norm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/doom_norm.png -------------------------------------------------------------------------------- /docs/07-advanced-topics/double-buffered.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/double-buffered.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/exporting-to-onnx.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/exporting-to-onnx.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/inactive-agents.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/inactive-agents.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/multi-policy-training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/multi-policy-training.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/normalization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/normalization.png -------------------------------------------------------------------------------- /docs/07-advanced-topics/normalizations.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/normalizations.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/observer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/observer.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/passing-info.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/passing-info.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/pbt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/pbt.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/policy-lag.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/policy-lag.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/profiling.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/profiling.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/serial-mode.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/serial-mode.md -------------------------------------------------------------------------------- /docs/07-advanced-topics/sync-async.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/07-advanced-topics/sync-async.md -------------------------------------------------------------------------------- /docs/08-miscellaneous/tests.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/08-miscellaneous/tests.md -------------------------------------------------------------------------------- /docs/08-miscellaneous/v1-to-v2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/08-miscellaneous/v1-to-v2.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/atari.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/atari.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/brax.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/brax.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/dmlab.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/dmlab.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/envpool.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/envpool.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/isaacgym.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/isaacgym.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/megaverse.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/megaverse.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/mujoco.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/mujoco.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/nethack.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/nethack.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/pettingzoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/pettingzoo.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/swarm-rl.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/swarm-rl.md -------------------------------------------------------------------------------- /docs/09-environment-integrations/vizdoom.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/09-environment-integrations/vizdoom.md -------------------------------------------------------------------------------- /docs/10-huggingface/huggingface.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/10-huggingface/huggingface.md -------------------------------------------------------------------------------- /docs/11-release-notes/release-notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/11-release-notes/release-notes.md -------------------------------------------------------------------------------- /docs/12-community/citation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/12-community/citation.md -------------------------------------------------------------------------------- /docs/12-community/contribution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/12-community/contribution.md -------------------------------------------------------------------------------- /docs/12-community/doc-contribution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/12-community/doc-contribution.md -------------------------------------------------------------------------------- /docs/CNAME: -------------------------------------------------------------------------------- 1 | www.samplefactory.dev 2 | -------------------------------------------------------------------------------- /docs/cfg-params.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/cfg-params.sh -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/docs/index.md -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/pyproject.toml -------------------------------------------------------------------------------- /sample_factory/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/evaluators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/evaluators/default_evaluator.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/learning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/learning/batcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/learning/batcher.py -------------------------------------------------------------------------------- /sample_factory/algo/learning/learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/learning/learner.py -------------------------------------------------------------------------------- /sample_factory/algo/learning/learner_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/learning/learner_worker.py -------------------------------------------------------------------------------- /sample_factory/algo/learning/rnn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/learning/rnn_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/runners/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/runners/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/runners/runner.py -------------------------------------------------------------------------------- /sample_factory/algo/runners/runner_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/runners/runner_parallel.py -------------------------------------------------------------------------------- /sample_factory/algo/runners/runner_serial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/runners/runner_serial.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/sampling/batched_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/batched_sampling.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/evaluation_sampling_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/evaluation_sampling_api.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/inference_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/inference_worker.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/non_batched_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/non_batched_sampling.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/rollout_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/rollout_worker.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/sampler.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/sampling_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/stats.py -------------------------------------------------------------------------------- /sample_factory/algo/sampling/sync_sampling_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/sampling/sync_sampling_api.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/utils/action_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/action_distributions.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/agent_policy_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/agent_policy_mapping.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/context.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/env_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/env_info.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/gymnasium_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/gymnasium_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/heartbeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/heartbeat.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/make_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/make_env.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/misc.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/model_sharing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/model_sharing.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/multiprocessing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/multiprocessing_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/optimizers.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/rl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/rl_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/running_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/running_mean_std.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/shared_buffers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/shared_buffers.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/spaces/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/algo/utils/spaces/discretized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/spaces/discretized.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/tensor_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/tensor_dict.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/tensor_utils.py -------------------------------------------------------------------------------- /sample_factory/algo/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/algo/utils/torch_utils.py -------------------------------------------------------------------------------- /sample_factory/benchmarking/benchmarking.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/benchmarking/benchmarking.txt -------------------------------------------------------------------------------- /sample_factory/benchmarking/benchmarking_isaacgym_ant.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/benchmarking/benchmarking_isaacgym_ant.txt -------------------------------------------------------------------------------- /sample_factory/benchmarking/benchmarking_xeon_36core.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/benchmarking/benchmarking_xeon_36core.txt -------------------------------------------------------------------------------- /sample_factory/cfg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/cfg/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/cfg/arguments.py -------------------------------------------------------------------------------- /sample_factory/cfg/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/cfg/cfg.py -------------------------------------------------------------------------------- /sample_factory/cfg/configurable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/cfg/configurable.py -------------------------------------------------------------------------------- /sample_factory/enjoy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/enjoy.py -------------------------------------------------------------------------------- /sample_factory/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/envs/create_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/envs/create_env.py -------------------------------------------------------------------------------- /sample_factory/envs/env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/envs/env_utils.py -------------------------------------------------------------------------------- /sample_factory/envs/env_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/envs/env_wrappers.py -------------------------------------------------------------------------------- /sample_factory/envs/pettingzoo_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/envs/pettingzoo_envs.py -------------------------------------------------------------------------------- /sample_factory/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/eval.py -------------------------------------------------------------------------------- /sample_factory/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/export_onnx.py -------------------------------------------------------------------------------- /sample_factory/huggingface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/huggingface/huggingface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/huggingface/huggingface_utils.py -------------------------------------------------------------------------------- /sample_factory/huggingface/load_from_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/huggingface/load_from_hub.py -------------------------------------------------------------------------------- /sample_factory/huggingface/push_to_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/huggingface/push_to_hub.py -------------------------------------------------------------------------------- /sample_factory/launcher/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/launcher/launcher_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/launcher_utils.py -------------------------------------------------------------------------------- /sample_factory/launcher/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/run.py -------------------------------------------------------------------------------- /sample_factory/launcher/run_description.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/run_description.py -------------------------------------------------------------------------------- /sample_factory/launcher/run_ngc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/run_ngc.py -------------------------------------------------------------------------------- /sample_factory/launcher/run_processes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/run_processes.py -------------------------------------------------------------------------------- /sample_factory/launcher/run_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/run_slurm.py -------------------------------------------------------------------------------- /sample_factory/launcher/slurm/sbatch_timeout.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/launcher/slurm/sbatch_timeout.sh -------------------------------------------------------------------------------- /sample_factory/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/model/action_parameterization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/action_parameterization.py -------------------------------------------------------------------------------- /sample_factory/model/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/actor_critic.py -------------------------------------------------------------------------------- /sample_factory/model/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/core.py -------------------------------------------------------------------------------- /sample_factory/model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/decoder.py -------------------------------------------------------------------------------- /sample_factory/model/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/encoder.py -------------------------------------------------------------------------------- /sample_factory/model/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/model_factory.py -------------------------------------------------------------------------------- /sample_factory/model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/model_utils.py -------------------------------------------------------------------------------- /sample_factory/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/model/utils.py -------------------------------------------------------------------------------- /sample_factory/pbt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/pbt/population_based_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/pbt/population_based_training.py -------------------------------------------------------------------------------- /sample_factory/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/train.py -------------------------------------------------------------------------------- /sample_factory/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sample_factory/utils/algo_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/algo_version.py -------------------------------------------------------------------------------- /sample_factory/utils/attr_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/attr_dict.py -------------------------------------------------------------------------------- /sample_factory/utils/decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/decay.py -------------------------------------------------------------------------------- /sample_factory/utils/dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/dicts.py -------------------------------------------------------------------------------- /sample_factory/utils/get_available_gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/get_available_gpus.py -------------------------------------------------------------------------------- /sample_factory/utils/gifs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/gifs.py -------------------------------------------------------------------------------- /sample_factory/utils/gpu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/gpu_utils.py -------------------------------------------------------------------------------- /sample_factory/utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/network.py -------------------------------------------------------------------------------- /sample_factory/utils/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/normalize.py -------------------------------------------------------------------------------- /sample_factory/utils/tb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/tb.py -------------------------------------------------------------------------------- /sample_factory/utils/timing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/timing.py -------------------------------------------------------------------------------- /sample_factory/utils/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/typing.py -------------------------------------------------------------------------------- /sample_factory/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/utils.py -------------------------------------------------------------------------------- /sample_factory/utils/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sample_factory/utils/wandb_utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/setup.py -------------------------------------------------------------------------------- /sf_examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/atari/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/atari/atari_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/atari/atari_params.py -------------------------------------------------------------------------------- /sf_examples/atari/atari_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/atari/atari_utils.py -------------------------------------------------------------------------------- /sf_examples/atari/enjoy_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/atari/enjoy_atari.py -------------------------------------------------------------------------------- /sf_examples/atari/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/atari/experiments/atari_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/atari/experiments/atari_envs.py -------------------------------------------------------------------------------- /sf_examples/atari/train_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/atari/train_atari.py -------------------------------------------------------------------------------- /sf_examples/brax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/brax/brax_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/brax/brax_render.py -------------------------------------------------------------------------------- /sf_examples/brax/enjoy_brax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/brax/enjoy_brax.py -------------------------------------------------------------------------------- /sf_examples/brax/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/brax/experiments/brax_basic_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/brax/experiments/brax_basic_envs.py -------------------------------------------------------------------------------- /sf_examples/brax/experiments/sbatch_timeout_brax.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/brax/experiments/sbatch_timeout_brax.sh -------------------------------------------------------------------------------- /sf_examples/brax/train_brax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/brax/train_brax.py -------------------------------------------------------------------------------- /sf_examples/dmlab/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/README.md -------------------------------------------------------------------------------- /sf_examples/dmlab/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/dmlab/command_line/train_dmlab30.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/command_line/train_dmlab30.sh -------------------------------------------------------------------------------- /sf_examples/dmlab/command_line/train_dmlab30_server_pbt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/command_line/train_dmlab30_server_pbt.sh -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab30.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab30.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_env.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_gym.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_level_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_level_cache.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_model.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_params.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_populate_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_populate_cache.py -------------------------------------------------------------------------------- /sf_examples/dmlab/dmlab_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/dmlab_utils.py -------------------------------------------------------------------------------- /sf_examples/dmlab/enjoy_dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/enjoy_dmlab.py -------------------------------------------------------------------------------- /sf_examples/dmlab/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/dmlab/experiments/dmlab30.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/experiments/dmlab30.py -------------------------------------------------------------------------------- /sf_examples/dmlab/train_dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/train_dmlab.py -------------------------------------------------------------------------------- /sf_examples/dmlab/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/dmlab/wrappers/reward_shaping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/dmlab/wrappers/reward_shaping.py -------------------------------------------------------------------------------- /sf_examples/enjoy_custom_env_custom_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/enjoy_custom_env_custom_model.py -------------------------------------------------------------------------------- /sf_examples/enjoy_custom_multi_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/enjoy_custom_multi_env.py -------------------------------------------------------------------------------- /sf_examples/enjoy_gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/enjoy_gym_env.py -------------------------------------------------------------------------------- /sf_examples/enjoy_pettingzoo_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/enjoy_pettingzoo_env.py -------------------------------------------------------------------------------- /sf_examples/envpool/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/envpool/atari/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/envpool/atari/enjoy_envpool_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/atari/enjoy_envpool_atari.py -------------------------------------------------------------------------------- /sf_examples/envpool/atari/envpool_atari_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/atari/envpool_atari_params.py -------------------------------------------------------------------------------- /sf_examples/envpool/atari/envpool_atari_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/atari/envpool_atari_utils.py -------------------------------------------------------------------------------- /sf_examples/envpool/atari/train_envpool_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/atari/train_envpool_atari.py -------------------------------------------------------------------------------- /sf_examples/envpool/envpool_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/envpool_utils.py -------------------------------------------------------------------------------- /sf_examples/envpool/envpool_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/envpool_wrappers.py -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/enjoy_envpool_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/mujoco/enjoy_envpool_mujoco.py -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/envpool_mujoco_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/mujoco/envpool_mujoco_params.py -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/envpool_mujoco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/mujoco/envpool_mujoco_utils.py -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/experiments/mujoco_envpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/mujoco/experiments/mujoco_envpool.py -------------------------------------------------------------------------------- /sf_examples/envpool/mujoco/train_envpool_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/envpool/mujoco/train_envpool_mujoco.py -------------------------------------------------------------------------------- /sf_examples/export_onnx_gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/export_onnx_gym_env.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/enjoy_isaacgym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/enjoy_isaacgym.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/isaacgym_allegrohand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/experiments/isaacgym_allegrohand.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/isaacgym_ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/experiments/isaacgym_ant.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/isaacgym_basic_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/experiments/isaacgym_basic_envs.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/isaacgym_humanoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/experiments/isaacgym_humanoid.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/experiments/isaacgym_runs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/experiments/isaacgym_runs.py -------------------------------------------------------------------------------- /sf_examples/isaacgym_examples/train_isaacgym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/isaacgym_examples/train_isaacgym.py -------------------------------------------------------------------------------- /sf_examples/mujoco/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/README.md -------------------------------------------------------------------------------- /sf_examples/mujoco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/mujoco/enjoy_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/enjoy_mujoco.py -------------------------------------------------------------------------------- /sf_examples/mujoco/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/mujoco/experiments/mujoco_all_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/experiments/mujoco_all_envs.py -------------------------------------------------------------------------------- /sf_examples/mujoco/fast_eval_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/fast_eval_mujoco.py -------------------------------------------------------------------------------- /sf_examples/mujoco/mujoco_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/mujoco_params.py -------------------------------------------------------------------------------- /sf_examples/mujoco/mujoco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/mujoco_utils.py -------------------------------------------------------------------------------- /sf_examples/mujoco/train_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/mujoco/train_mujoco.py -------------------------------------------------------------------------------- /sf_examples/nethack/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/nethack/enjoy_nethack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/enjoy_nethack.py -------------------------------------------------------------------------------- /sf_examples/nethack/fast_eval_nethack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/fast_eval_nethack.py -------------------------------------------------------------------------------- /sf_examples/nethack/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/models/__init__.py -------------------------------------------------------------------------------- /sf_examples/nethack/models/chaotic_dwarf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/models/chaotic_dwarf.py -------------------------------------------------------------------------------- /sf_examples/nethack/models/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/models/crop.py -------------------------------------------------------------------------------- /sf_examples/nethack/models/scaled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/models/scaled.py -------------------------------------------------------------------------------- /sf_examples/nethack/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/models/utils.py -------------------------------------------------------------------------------- /sf_examples/nethack/nethack_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/nethack_env.py -------------------------------------------------------------------------------- /sf_examples/nethack/nethack_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/nethack_params.py -------------------------------------------------------------------------------- /sf_examples/nethack/scripts/sample_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/scripts/sample_env.py -------------------------------------------------------------------------------- /sf_examples/nethack/train_nethack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/train_nethack.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/nethack/utils/task_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/task_rewards.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/__init__.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/blstats_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/blstats_info.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/no_progress_timeout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/no_progress_timeout.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/prev_actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/prev_actions.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/task_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/task_rewards.py -------------------------------------------------------------------------------- /sf_examples/nethack/utils/wrappers/tile_tty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/nethack/utils/wrappers/tile_tty.py -------------------------------------------------------------------------------- /sf_examples/notebooks/samplefactory_hub_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/notebooks/samplefactory_hub_example.ipynb -------------------------------------------------------------------------------- /sf_examples/sampler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/sampler/use_simplified_sampling_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/sampler/use_simplified_sampling_api.py -------------------------------------------------------------------------------- /sf_examples/train_custom_env_custom_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/train_custom_env_custom_model.py -------------------------------------------------------------------------------- /sf_examples/train_custom_multi_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/train_custom_multi_env.py -------------------------------------------------------------------------------- /sf_examples/train_gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/train_gym_env.py -------------------------------------------------------------------------------- /sf_examples/train_pettingzoo_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/train_pettingzoo_env.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/custom_env/custom_doom_env.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/custom_env/custom_doom_env.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/custom_env/custom_doom_env.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/custom_env/custom_doom_env.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/README.md -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/action_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/action_space.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_gym.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_model.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_params.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_play_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_play_demo.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_render.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/doom_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/doom_utils.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/multi_agent_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/multi_agent_match.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/multiplayer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/multiplayer/doom_multiagent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/multiplayer/doom_multiagent.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/multiplayer/doom_multiagent_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/multiplayer/doom_multiagent_wrapper.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/play_doom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/play_doom.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/sample_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/sample_env.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/basic.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/basic.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/basic.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/basic.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/battle.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/battle.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/battle.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/battle.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/battle2.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/battle2.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/battle2_continuous_turning.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/battle2_continuous_turning.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/battle_continuous_turning.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/battle_continuous_turning.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/cig.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/cig.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/cig.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/cig.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/deadly_corridor.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/deadly_corridor.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/deadly_corridor.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/deadly_corridor.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/defend_the_center.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/defend_the_center.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/defend_the_center.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/defend_the_center.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/defend_the_line.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/defend_the_line.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/defend_the_line.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/defend_the_line.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/dwango5.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/dwango5.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/dwango5_dm.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/dwango5_dm.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/dwango5_dm_continuous_weap.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/dwango5_dm_continuous_weap.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/freedm.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/freedm.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/freedm.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/freedm.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/health_gathering.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/health_gathering.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/health_gathering.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/health_gathering.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/health_gathering_supreme.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/health_gathering_supreme.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/health_gathering_supreme.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/health_gathering_supreme.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/my_way_home.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/my_way_home.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/my_way_home.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/my_way_home.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/ssl2.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/ssl2.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/ssl2.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/ssl2.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/two_colors_easy.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/two_colors_easy.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/two_colors_easy.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/two_colors_easy.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/two_colors_hard.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/two_colors_hard.cfg -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scenarios/two_colors_hard.wad: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scenarios/two_colors_hard.wad -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/scripts/generate_bots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/scripts/generate_bots.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/additional_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/additional_input.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/bot_difficulty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/bot_difficulty.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/exploration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/exploration.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/multiplayer_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/multiplayer_stats.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/observation_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/observation_space.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/reward_shaping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/reward_shaping.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/scenario_wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/scenario_wrappers/gathering_reward_shaping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/scenario_wrappers/gathering_reward_shaping.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/doom/wrappers/step_human_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/doom/wrappers/step_human_input.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/enjoy_custom_vizdoom_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/enjoy_custom_vizdoom_env.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/enjoy_vizdoom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/enjoy_vizdoom.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_basic.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_battle_battle2_appo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_battle_battle2_appo.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_battle_d4_appo_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_battle_d4_appo_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_bots_sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_bots_sweep.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_defend_center.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_defend_center.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_freedm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_freedm.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/doom_health_gathering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/doom_health_gathering.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_all_basic_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_all_basic_envs.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_battle2_appo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_battle2_appo.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_battle2_appo_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_battle2_appo_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_battle_appo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_battle_appo.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_battle_appo_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_battle_appo_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_bots_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_bots_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_duel_bots_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_duel_bots_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_duel_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_duel_pbt.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/paper_doom_wall_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/paper_doom_wall_time.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/experiments/sf2_doom_battle_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/experiments/sf2_doom_battle_envs.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/train_custom_vizdoom_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/train_custom_vizdoom_env.py -------------------------------------------------------------------------------- /sf_examples/vizdoom/train_vizdoom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/sf_examples/vizdoom/train_vizdoom.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/algo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/algo/test_action_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_action_distributions.py -------------------------------------------------------------------------------- /tests/algo/test_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_learner.py -------------------------------------------------------------------------------- /tests/algo/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_model.py -------------------------------------------------------------------------------- /tests/algo/test_model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_model_builder.py -------------------------------------------------------------------------------- /tests/algo/test_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_pbt.py -------------------------------------------------------------------------------- /tests/algo/test_rms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_rms.py -------------------------------------------------------------------------------- /tests/algo/test_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_rnn.py -------------------------------------------------------------------------------- /tests/algo/test_spaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_spaces.py -------------------------------------------------------------------------------- /tests/algo/test_tensordict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/algo/test_tensordict.py -------------------------------------------------------------------------------- /tests/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/atari/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/atari/test_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/atari/test_atari.py -------------------------------------------------------------------------------- /tests/envs/atari/test_envpool_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/atari/test_envpool_atari.py -------------------------------------------------------------------------------- /tests/envs/dmlab/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/dmlab/test_dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/dmlab/test_dmlab.py -------------------------------------------------------------------------------- /tests/envs/mujoco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/mujoco/test_envpool_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/mujoco/test_envpool_mujoco.py -------------------------------------------------------------------------------- /tests/envs/mujoco/test_mujoco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/mujoco/test_mujoco.py -------------------------------------------------------------------------------- /tests/envs/pettingzoo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/pettingzoo/test_pettingzoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/pettingzoo/test_pettingzoo.py -------------------------------------------------------------------------------- /tests/envs/tuple_action_envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/tuple_action_envs/test_mixed_dist_env_batched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/tuple_action_envs/test_mixed_dist_env_batched.py -------------------------------------------------------------------------------- /tests/envs/tuple_action_envs/test_mixed_dist_env_non_batched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/tuple_action_envs/test_mixed_dist_env_non_batched.py -------------------------------------------------------------------------------- /tests/envs/tuple_action_envs/test_two_discrete_action_dist_env_batched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/tuple_action_envs/test_two_discrete_action_dist_env_batched.py -------------------------------------------------------------------------------- /tests/envs/tuple_action_envs/test_two_discrete_action_dist_env_non_batched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/tuple_action_envs/test_two_discrete_action_dist_env_non_batched.py -------------------------------------------------------------------------------- /tests/envs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/utils.py -------------------------------------------------------------------------------- /tests/envs/vizdoom/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/envs/vizdoom/test_vizdoom_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/vizdoom/test_vizdoom_basic.py -------------------------------------------------------------------------------- /tests/envs/vizdoom/test_vizdoom_multiplayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/envs/vizdoom/test_vizdoom_multiplayer.py -------------------------------------------------------------------------------- /tests/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/examples/test_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/examples/test_example.py -------------------------------------------------------------------------------- /tests/examples/test_example_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/examples/test_example_multi.py -------------------------------------------------------------------------------- /tests/examples/test_example_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/examples/test_example_sampler.py -------------------------------------------------------------------------------- /tests/export_onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/export_onnx_utils.py -------------------------------------------------------------------------------- /tests/test_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/test_launcher.py -------------------------------------------------------------------------------- /tests/test_precheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/test_precheck.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alex-petrenko/sample-factory/HEAD/tests/utils.py --------------------------------------------------------------------------------