Functions — github.com/Theohhhu/UPDeT

CodeHub MCPcopy Create free account

hub / github.com/Theohhhu/UPDeT / functions

Functions148 in github.com/Theohhhu/UPDeT

⨍ Functions 148◇ Types & classes 29

src/runners/parallel_runner.py:259

(self, args, logger)

src/runners/episode_runner.py:9

(self, mac, scheme, logger, args)

src/learners/qtran_learner.py:9

(self, mac, scheme, logger, args)

src/learners/q_learner.py:10

(self, mac, scheme, logger, args)

src/learners/coma_learner.py:10

src/components/episode_buffer.py:200

src/components/episode_buffer.py:243

Method__setstate__

src/runners/parallel_runner.py:264

Methodclose_env

src/runners/parallel_runner.py:57

Functionconvert

src/utils/dict2namedtuple.py:4

src/learners/qtran_learner.py:161

src/learners/coma_learner.py:154

(env, **kwargs)

src/envs/__init__.py:6

Functionenv_worker

(remote, env_fn)

src/runners/parallel_runner.py:213

(self, inputs, hidden_state, task_enemy_num, task_ally_num)

src/modules/agents/transformer_agg_agent.py:18

(self, x, mask)

src/modules/agents/transformer_agg_agent.py:44

src/modules/agents/transformer_agg_agent.py:105

(self, inputs, hidden_state)

src/modules/agents/rnn_agent.py:18

(self, x, mask)

src/modules/agents/updet_agent.py:57

src/modules/agents/updet_agent.py:118

(self, x, h, mask)

src/modules/agents/updet_agent.py:154

(self, batch, t=None)

src/modules/critics/coma.py:22

(self, agent_qs, batch)

src/modules/mixers/vdn.py:9

(self, batch, hidden_states, actions=None)

src/modules/mixers/qtran.py:81

Methodget_avail_agent_actions

Returns the available actions for agent_id

src/envs/multiagentenv.py:29

Methodget_env_info

src/runners/episode_runner.py:34

Methodget_obs_agent

Returns observation for agent_id

src/envs/multiagentenv.py:11

Methodinfer_output_info

(self, vshape_in, dtype_in)

src/components/transforms.py:8

Methodinit_hidden

(self, batch_size)

src/controllers/basic_controller.py:66

Methodinit_hidden

src/modules/agents/rnn_agent.py:14

Methodload_models

src/controllers/basic_controller.py:85

Methodload_models

src/learners/qtran_learner.py:174

Methodload_models

src/learners/coma_learner.py:165

Functionmy_main

(_run, _config, _log)

Functionprint_time

(start_time, T, t_max, episode, episode_rewards)

src/utils/timehelper.py:5

src/envs/multiagentenv.py:42

Returns initial observations and states

src/envs/multiagentenv.py:38

(self, test_mode=False)

src/runners/parallel_runner.py:85

Methodsave_models

src/controllers/basic_controller.py:82

Methodsave_models

src/learners/qtran_learner.py:168

Methodsave_models

src/learners/coma_learner.py:159

Methodsave_replay

src/runners/parallel_runner.py:54

Methodsave_replay

src/runners/episode_runner.py:37

Methodselect_action

(self, agent_inputs, avail_actions, t_env, test_mode=False)

src/components/action_selectors.py:44

(self, scheme, groups, preprocess, mac)

src/runners/parallel_runner.py:43

(self, batch: EpisodeBatch, t_env: int, episode_num: int)

src/learners/qtran_learner.py:34

(self, batch: EpisodeBatch, t_env: int, episode_num: int)

src/learners/coma_learner.py:32

Methodtransform

src/components/transforms.py:5

← previous101–148 of 148, ranked by callers