Functions — github.com/Theohhhu/UPDeT

hub / github.com/Theohhhu/UPDeT / functions

Functions148 in github.com/Theohhhu/UPDeT

⨍ Functions 148◇ Types & classes 29

↓ 29 callersMethodlog_stat

(self, key, value, t, to_sacred=True)

src/utils/logging.py:26

↓ 13 callersMethodupdate

(self, data, bs=slice(None), ts=slice(None), mark_filled=True)

src/components/episode_buffer.py:87

↓ 9 callersMethodcuda

(self)

src/learners/q_learner.py:124

↓ 8 callersMethodinit_hidden

(self)

src/modules/agents/updet_agent.py:14

↓ 6 callersMethodparameters

(self)

src/controllers/basic_controller.py:73

↓ 6 callersMethodstep

Returns reward, terminated, info

src/envs/multiagentenv.py:3

↓ 5 callersMethodforward

(self, agent_qs, states)

src/modules/mixers/qmix.py:41

↓ 4 callersMethodeval

(self, T)

src/components/epsilon_schedules.py:21

↓ 4 callersMethodget_avail_actions

(self)

src/envs/multiagentenv.py:26

↓ 4 callersMethodget_obs

Returns all agent observations in a list

src/envs/multiagentenv.py:7

↓ 4 callersMethodget_state

(self)

src/envs/multiagentenv.py:19

↓ 4 callersFunctiontime_str

Convert seconds to a nicer string showing days, hours, minutes and seconds

src/utils/timehelper.py:28

↓ 3 callersMethod__init__

(self, input_dim, emb, heads, depth, output_dim)

src/modules/agents/transformer_agg_agent.py:125

↓ 3 callersMethod__init__

(self, input_shape, args)

src/modules/agents/updet_agent.py:8

↓ 3 callersMethodclose

(self)

src/envs/multiagentenv.py:45

↓ 3 callersMethodrun

(self, test_mode=False)

src/runners/episode_runner.py:48

↓ 3 callersMethodsample

(self, batch_size)

src/components/episode_buffer.py:234

↓ 3 callersMethodselect_actions

(self, ep_batch, t_ep, t_env, bs=slice(None), test_mode=False)

src/controllers/basic_controller.py:19

↓ 2 callersFunction_get_config

(params, arg_name, subfolder)

src/main.py:38

↓ 2 callersMethod_get_num_items

(self, indexing_item, max_size)

src/components/episode_buffer.py:161

↓ 2 callersMethod_log

(self, returns, stats, prefix)

src/runners/parallel_runner.py:202

↓ 2 callersMethod_log

(self, returns, stats, prefix)

src/runners/episode_runner.py:115

↓ 2 callersMethod_new_data_sn

(self)

src/components/episode_buffer.py:168

↓ 2 callersMethod_parse_slices

(self, items)

src/components/episode_buffer.py:174

↓ 2 callersMethod_setup_data

(self, scheme, groups, batch_size, max_seq_length, preprocess)

src/components/episode_buffer.py:30

↓ 2 callersMethodcan_sample

(self, batch_size)

src/components/episode_buffer.py:231

↓ 2 callersMethodclose_env

(self)

src/runners/episode_runner.py:40

↓ 2 callersMethodforward

(self, x, h, mask)

src/modules/agents/transformer_agg_agent.py:141

↓ 2 callersMethodload_state

(self, other_mac)

src/controllers/basic_controller.py:76

↓ 2 callersFunctionrecursive_dict_update

(d, u)

src/main.py:55

↓ 2 callersMethodreset

(self)

src/runners/parallel_runner.py:61

↓ 2 callersMethodto

(self, device)

src/components/episode_buffer.py:80

↓ 1 callersMethod__init__

(self, scheme, groups, batch_size, max_seq

src/components/episode_buffer.py:7

↓ 1 callersMethod_build_agents

(self, input_shape)

src/controllers/basic_controller.py:88

↓ 1 callersMethod_build_inputs

(self, batch, t)

src/controllers/basic_controller.py:91

↓ 1 callersMethod_build_inputs

(self, batch, t=None)

src/modules/critics/coma.py:29

↓ 1 callersMethod_build_inputs_transformer

(self, batch, t)

src/controllers/basic_controller.py:108

↓ 1 callersMethod_check_safe_view

(self, v, dest)

src/components/episode_buffer.py:114

↓ 1 callersMethod_get_input_shape

(self, scheme)

src/controllers/basic_controller.py:119

↓ 1 callersMethod_get_input_shape

(self, scheme)

src/modules/critics/coma.py:61

↓ 1 callersMethod_train_critic

(self, batch, rewards, terminated, actions, avail_actions, mask, bs, max_t)

src/learners/coma_learner.py:100

↓ 1 callersMethod_update_targets

(self)

src/learners/qtran_learner.py:155

↓ 1 callersMethod_update_targets

(self)

src/learners/q_learner.py:118

↓ 1 callersMethod_update_targets

(self)

src/learners/coma_learner.py:150

↓ 1 callersFunctionargs_sanity_check

(config, _log)

src/run.py:222

↓ 1 callersFunctionbuild_td_lambda_targets

(rewards, terminated, mask, target_qs, n_agents, gamma, td_lambda)

src/utils/rl_utils.py:4

↓ 1 callersFunctionconfig_copy

(config)

src/main.py:64

↓ 1 callersMethodcuda

(self)

src/controllers/basic_controller.py:79

↓ 1 callersFunctionevaluate_sequential

(args, runner)

src/run.py:72

↓ 1 callersMethodextend

(self, scheme, groups=None)

src/components/episode_buffer.py:77

↓ 1 callersMethodforward

(self, ep_batch, t, test_mode=False)

src/controllers/basic_controller.py:26

↓ 1 callersMethodforward

(self, inputs, hidden_state, task_enemy_num, task_ally_num)

src/modules/agents/updet_agent.py:18

↓ 1 callersMethodget_env_info

(self)

src/runners/parallel_runner.py:51

↓ 1 callersMethodget_env_info

(self)

src/envs/multiagentenv.py:54

↓ 1 callersFunctionget_logger

()

src/utils/logging.py:55

↓ 1 callersMethodget_obs_size

Returns the shape of the observation

src/envs/multiagentenv.py:15

↓ 1 callersMethodget_state_size

Returns the shape of the state

src/envs/multiagentenv.py:22

↓ 1 callersMethodget_total_actions

Returns the total number of actions an agent could ever take

src/envs/multiagentenv.py:33

↓ 1 callersMethodinfer_output_info

(self, vshape_in, dtype_in)

src/components/transforms.py:21

↓ 1 callersMethodinit_hidden

(self)

src/modules/agents/transformer_agg_agent.py:14

↓ 1 callersMethodinsert_episode_batch

(self, ep_batch)

src/components/episode_buffer.py:214

↓ 1 callersMethodload_models

(self, path)

src/learners/q_learner.py:137

↓ 1 callersFunctionmask_

(matrices, maskval=0.0, mask_diagonal=True)

src/modules/agents/transformer_agg_agent.py:154

↓ 1 callersFunctionmask_

(matrices, maskval=0.0, mask_diagonal=True)

src/modules/agents/updet_agent.py:167

↓ 1 callersMethodmax_t_filled

(self)

src/components/episode_buffer.py:197

↓ 1 callersMethodprint_recent_stats

(self)

src/utils/logging.py:40

↓ 1 callersMethodreset

(self)

src/runners/episode_runner.py:43

↓ 1 callersFunctionrun

(_run, _config, _log)

src/run.py:19

↓ 1 callersFunctionrun_sequential

(args, logger)

src/run.py:82

↓ 1 callersMethodsave_models

(self, path)

src/learners/q_learner.py:131

↓ 1 callersMethodsave_replay

(self)

src/envs/multiagentenv.py:51

↓ 1 callersMethodseed

(self)

src/envs/multiagentenv.py:48

↓ 1 callersMethodselect_action

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

src/components/action_selectors.py:18

↓ 1 callersMethodsetup

(self, scheme, groups, preprocess, mac)

src/runners/episode_runner.py:29

↓ 1 callersMethodsetup_sacred

(self, sacred_run_dict)

src/utils/logging.py:22

↓ 1 callersMethodsetup_tb

(self, directory_name)

src/utils/logging.py:15

↓ 1 callersFunctiontime_left

(start_time, t_start, t_current, t_max)

src/utils/timehelper.py:17

↓ 1 callersMethodtrain

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

src/learners/q_learner.py:37

↓ 1 callersMethodtransform

(self, tensor)

src/components/transforms.py:16

Method__getitem__

(self, item)

src/components/episode_buffer.py:123

Method__getstate__

(self)

src/runners/parallel_runner.py:261

Method__init__

(self, scheme, groups, args)

src/controllers/basic_controller.py:8

Method__init__

(self, input_shape, args)

src/modules/agents/transformer_agg_agent.py:8

Method__init__

(self, emb, heads=8, mask=False)

src/modules/agents/transformer_agg_agent.py:30

Method__init__

(self, emb, heads, mask, ff_hidden_mult=4, dropout=0.0)

src/modules/agents/transformer_agg_agent.py:88

Method__init__

(self, input_shape, args)

src/modules/agents/rnn_agent.py:6

Method__init__

(self, emb, heads=8, mask=False)

src/modules/agents/updet_agent.py:43

Method__init__

(self, emb, heads, mask, ff_hidden_mult=4, dropout=0.0)

src/modules/agents/updet_agent.py:101

Method__init__

(self, input_dim, emb, heads, depth, output_dim)

src/modules/agents/updet_agent.py:138

Method__init__

(self, scheme, args)

src/modules/critics/coma.py:7

Method__init__

(self)

src/modules/mixers/vdn.py:6

Method__init__

(self, args)

src/modules/mixers/qtran.py:8

Method__init__

(self, args)

src/modules/mixers/qmix.py:8

Method__init__

(self, args)

src/components/action_selectors.py:10

Method__init__

(self, args)

src/components/action_selectors.py:37

Method__init__

(self, start, finish, time_length, decay="

src/components/epsilon_schedules.py:6

Method__init__

(self, out_dim)

src/components/transforms.py:13

Method__init__

(self, scheme, groups, buffer_size, max_seq_length, preprocess=None, device="cpu")

src/components/episode_buffer.py:208

Method__init__

(self, console_logger)

src/utils/logging.py:6

Method__init__

(self, args, logger)

src/runners/parallel_runner.py:13

next →1–100 of 148, ranked by callers