Functions — github.com/YushuoLi/Gato-A-Generalist-Agent

CodeHub MCPcopy Create free account

hub / github.com/YushuoLi/Gato-A-Generalist-Agent / functions

Functions78 in github.com/YushuoLi/Gato-A-Generalist-Agent

⨍ Functions 78 ◇ Types & classes 18

↓ 5 callersMethod__init__

Gato_models/nets.py:156

↓ 5 callersMethodforward

Gato_models/nets.py:159

↓ 3 callersMethod__init__

Appendix C.2. Embedding Function

Gato_models/embedding.py:138

↓ 3 callersMethod__init__

(self, state_dim, act_dim, n_blocks, context_len, n_heads, drop_p, config: U

Gato_models/model.py:170

↓ 3 callersFunctionget_d4rl_normalized_score

归一化某环境的得分 :param score: 得分 :param env_name: 环境名 :return:

data/data_loader.py:31

↓ 2 callersFunction_broadcast

(row_pos, col_pos, row_ones, col_ones)

Gato_models/embedding.py:31

↓ 2 callersMethod_discretize

Gato_models/embedding.py:84

↓ 2 callersMethod_discretize_interval

(self, axis_num)

Gato_models/embedding.py:87

↓ 2 callersFunction_randomized_positions

(from_v, to_v, device)

Gato_models/embedding.py:13

↓ 2 callersFunction_rounded_mean_positions

Gato_models/embedding.py:24

↓ 2 callersFunctiondiscount_cumsum

计算returns-to-go :param x: 回报序列 :param gamma: 折扣系数 :return:

data/data_loader.py:17

↓ 1 callersMethodbuild

Gato_models/embedding.py:97

↓ 1 callersFunctioncheck_atari_dataloader

测试数据库的数据格式

data/data_loader.py:316

↓ 1 callersFunctioncheck_savefile

检查保存的数据文件的格式

data/download_d4rl_datasets.py:69

↓ 1 callersFunctioncollect_atari_data

data/collect_atari_data.py:23

↓ 1 callersFunctionevaluate_dt

(model, device, context_len, env, rtg_target, rtg_scale, num_eval_ep=10, max_test_ep_len=1

utils/utils.py:12

↓ 1 callersMethodevaluate_expert

data/data_loader.py:133

↓ 1 callersFunctionevaluate_gato

(model, device, context_len, env, rtg_scale, eval_data, num_eval_ep=1, max

utils/utils.py:102

↓ 1 callersFunctionget_d4rl_dataset_stats

获得某环境的状态均值、标准差信息 :param env_d4rl_name: 环境名 :return:

data/data_loader.py:43

↓ 1 callersMethodget_state_stats

data/data_loader.py:83

↓ 1 callersMethodget_state_stats

data/data_loader.py:230

↓ 1 callersFunctioninverse_tokenize_act_values

(x, bins=1024, act_continuous=True, shift=33024)

Gato_models/tokenizers.py:46

↓ 1 callersFunctionload_dataset

(env_name, dataset_type)

data/test_d4rl_dataset.py:21

↓ 1 callersFunctionmu_law_encode

(x, mu=100, m=256)

Gato_models/tokenizers.py:9

↓ 1 callersFunctionplot

utils/plot.py:7

↓ 1 callersMethodprompt

(self, ori_traj)

data/data_loader.py:172

↓ 1 callersMethodsequence

(self, timesteps, states, actions)

Gato_models/model.py:213

↓ 1 callersFunctiontest

↓ 1 callersFunctiontokenize_act_values

(x, bins=1024, act_continuous=True, shift=33024)

Gato_models/tokenizers.py:32

↓ 1 callersFunctiontokenize_obs_values

(x, mu=100, m=256, bins=1024, obs_continuous=True, shift=32000)

Gato_models/tokenizers.py:18

↓ 1 callersFunctiontrain

run/train.py:21

↓ 1 callersFunctiontrain

run/check_train.py:25

Method__getitem__

data/data_loader.py:89

Method__getitem__

data/data_loader.py:236

(self, file_name="Default.log", stream=sys.stdout)

utils/logger.py:43

(self, dataset_path, context_len, rtg_scale, prompt_symbol=True, prompt_length=40)

data/data_loader.py:56

(self, dataset_path, context_len, rtg_scale)

data/data_loader.py:199

(self, **kwargs)

Gato_models/config.py:31

(self, in_channels: int, out_channels: int, stride: int)

Gato_models/nets.py:9

(self, in_channels: int, out_channels: int, stride: int)

Gato_models/nets.py:22

(self, in_channels: int, bottleneck_channels: int, out_channels: int, stride: int)

Gato_models/nets.py:50

(self, n_blocks: List[int], n_channels: List[int], bottlenecks: Optional[List[int]] = None,

Gato_models/nets.py:83

(self, in_channels: int, out_channels: int, stride: int, num_group_norm_groups)

Gato_models/nets.py:132

Appendix C.3. Position Encodings

Gato_models/embedding.py:52

Appendix C.3. Position Encodings > Local Observation Position Encodings

Gato_models/embedding.py:185

(self, config: Union[GatoConfig, Dict[str, Any]])

Gato_models/embedding.py:216

(self, h_dim, max_T, n_heads, drop_p)

Gato_models/model.py:15

(self, h_dim, max_T, n_heads, drop_p)

Gato_models/model.py:67

(self, state_dim, act_dim, n_blocks, h_dim, context_len, n_heads, drop_p, max_timestep=4096,

Gato_models/model.py:92

data/data_loader.py:86

data/data_loader.py:233

Gato_models/config.py:16

Functioncheck_d4rl_dataloader

测试数据库的数据格式

data/data_loader.py:276

Functiondownload_d4rl_data

data/download_d4rl_datasets.py:13

Methodembedding_input_size

Gato_models/config.py:72

utils/logger.py:51

(self, x: torch.Tensor)

Gato_models/nets.py:16

(self, x: torch.Tensor)

Gato_models/nets.py:40

(self, x: torch.Tensor)

Gato_models/nets.py:72

(self, x: torch.Tensor)

Gato_models/nets.py:122

(self, x: torch.Tensor)

Gato_models/nets.py:146

Gato_models/embedding.py:126

Gato_models/embedding.py:166

(self, inputs, sym_index)

Gato_models/embedding.py:199

Gato_models/embedding.py:229

Gato_models/model.py:37

Gato_models/model.py:79

(self, timesteps, states, actions, returns_to_go)

Gato_models/model.py:129

(self, timesteps, states, actions)

Gato_models/model.py:276

Methodfrom_dict

(cls, config_dict: Dict[str, Any])

Gato_models/config.py:86

Gato_models/config.py:8

Functionload_dataset_all

data/test_d4rl_dataset.py:8

Functionlogger_1

:param name: 日志收集器名字 :param level: 日志收集器的等级 :param filename: 日志文件的名称 :param sh_level: 控制台输出日志的等级 :param fh_level: 文件输出日志的等级

utils/logger.py:9

Functionlogger_2

(log_path, file_name)

utils/logger.py:54

Methodoutput_target_size

Gato_models/config.py:77

Gato_models/config.py:24

Gato_models/config.py:81

(self, message)

utils/logger.py:47