MCPcopy Create free account

hub / github.com/charleshsc/QT / functions

Functions136 in github.com/charleshsc/QT

↓ 19 callersMethodrecord_tabular
(self, key, val)
logger.py:361
↓ 14 callersMethodlog
(self, s, with_prefix=True, with_timestamp=True)
logger.py:345
↓ 10 callersMethodstep
(self, action)
decision_transformer/envs/reacher_2d.py:22
↓ 7 callersMethodforward
(self, state, action)
decision_transformer/models/ql_DT.py:30
↓ 5 callersMethod__init__
(self, n_state, config)
decision_transformer/models/trajectory_gpt2.py:248
↓ 5 callersFunction_strip_invisible
Remove invisible ANSI color codes.
tabulate.py:341
↓ 4 callersFunction_build_line
Return a string which represents a horizontal line.
tabulate.py:796
↓ 4 callersFunction_build_row
Return a string which represents a row of data cells.
tabulate.py:786
↓ 3 callersMethod_add_output
(self, file_name, arr, fds, mode='a')
logger.py:285
↓ 3 callersMethodsplit_heads
(self, x, k=False)
decision_transformer/models/trajectory_gpt2.py:194
↓ 2 callersFunction_build_simple_row
Format row according to DataRow format without padding.
tabulate.py:780
↓ 2 callersMethod_get_obs
(self)
decision_transformer/envs/reacher_2d.py:36
↓ 2 callersFunction_isconvertible
(conv, string)
tabulate.py:212
↓ 2 callersFunction_isint
>>> _isint("123") True >>> _isint("123.45") False
tabulate.py:232
↓ 2 callersFunction_isnumber
>>> _isnumber("123.45") True >>> _isnumber("123") True >>> _isnumber("spam") False
tabulate.py:220
↓ 2 callersFunction_pad_row
(cells, padding)
tabulate.py:808
↓ 2 callersMethod_remove_output
(self, file_name, arr, fds)
logger.py:291
↓ 2 callersMethoddump_tabular
(self, *args, **kwargs)
logger.py:441
↓ 2 callersMethodget_action
(self, states, actions, rewards, **kwargs)
decision_transformer/models/model.py:19
↓ 2 callersFunctionmkdir_p
(path)
logger.py:247
↓ 2 callersMethodpop_tabular_prefix
(self, )
logger.py:376
↓ 2 callersMethodpush_prefix
(self, prefix)
logger.py:297
↓ 2 callersMethodpush_tabular_prefix
(self, key)
logger.py:372
↓ 2 callersMethodq_min
(self, state, action)
decision_transformer/models/ql_DT.py:38
↓ 2 callersMethodreset
(self)
logger.py:282
↓ 2 callersFunctiontabulate
Format a fixed width table for pretty printing. >>> print(tabulate([[1, 2.34], [-56, "8.999"], ["2", "10001"]])) --- --------- 1
tabulate.py:541
↓ 1 callersMethod__init__
(self)
logger.py:258
↓ 1 callersMethod__init__
(self, state_dim, action_dim, hidden_dim=256)
decision_transformer/models/ql_DT.py:12
↓ 1 callersFunction_afterpoint
Symbols after a decimal point, -1 if the string lacks the decimal point. >>> _afterpoint("123.45") 2 >>> _afterpoint("1001") -1 >
tabulate.py:278
↓ 1 callersFunction_align_column
[string] -> [padded_string] >>> list(map(str,_align_column(["12.345", "-1234.5", "1.23", "1234.5", "1e+234", "1.0e234"], "decimal"))) [' 12
tabulate.py:362
↓ 1 callersFunction_align_header
(header, alignment, width)
tabulate.py:456
↓ 1 callersMethod_attn
(self, q, k, v, attention_mask=None, head_mask=None, output_attentions=False)
decision_transformer/models/trajectory_gpt2.py:162
↓ 1 callersFunction_format
Format a value accoding to its type. Unicode is supported: >>> hrow = ['\u0431\u0443\u043a\u0432\u0430', '\u0446\u0438\u0444\u0440\u0430'] ;
tabulate.py:431
↓ 1 callersFunction_format_table
Produce a plain-text representation of the table.
tabulate.py:817
↓ 1 callersFunction_normalize_tabular_data
Transform a supported data type to a list of lists, and a list of headers. Supported tabular data types: * list-of-lists or another iterable
tabulate.py:467
↓ 1 callersFunction_padboth
Center string. >>> _padboth(6, '\u044f\u0439\u0446\u0430') == ' \u044f\u0439\u0446\u0430 ' True
tabulate.py:329
↓ 1 callersFunction_padleft
Flush right. >>> _padleft(6, '\u044f\u0439\u0446\u0430') == ' \u044f\u0439\u0446\u0430' True
tabulate.py:305
↓ 1 callersFunction_padright
Flush left. >>> _padright(6, '\u044f\u0439\u0446\u0430') == '\u044f\u0439\u0446\u0430 ' True
tabulate.py:317
↓ 1 callersFunction_pipe_segment_with_colons
Return a segment of a horizontal line with optional colons which indicate column's alignment (as in `pipe` output format).
tabulate.py:78
↓ 1 callersFunction_type
The least generic type (type(None), int, float, str, unicode). >>> _type(None) is type(None) True >>> _type("foo") is type("") True
tabulate.py:244
↓ 1 callersMethodadd_tabular_output
(self, file_name, relative_to_snapshot_dir=False)
logger.py:308
↓ 1 callersMethodadd_text_output
(self, file_name)
logger.py:301
↓ 1 callersFunctioncreate_exp_name
Create a semi-unique experiment name that has a timestamp :param exp_prefix: :param exp_id: :return:
logger.py:55
↓ 1 callersFunctioncreate_log_dir
Creates and returns a unique log directory. :param exp_prefix: All experiments with this prefix will have log directories be under this d
logger.py:66
↓ 1 callersFunctiondict_to_safe_json
Convert each value in the dictionary into a JSON'able primitive. :param d: :return:
logger.py:26
↓ 1 callersFunctiondiscount_cumsum
(x, gamma)
experiment.py:48
↓ 1 callersFunctioneval_episodes
(target_rew)
experiment.py:298
↓ 1 callersFunctionevaluate_episode_rtg
( env, state_dim, act_dim, model, critic, max_ep_len=2000,
decision_transformer/evaluation/evaluate_episodes.py:68
↓ 1 callersFunctionexperiment
( exp_prefix, variant, )
experiment.py:64
↓ 1 callersMethodforward
(self, states, actions, rewards=None, targets=None, returns_to_go=None, timesteps=None, attention_mask=None)
decision_transformer/models/decision_transformer.py:57
↓ 1 callersMethodforward
(self, states, actions, rewards=None, targets=None, returns_to_go=None, timesteps=None, attention_mask=None)
decision_transformer/models/ql_DT.py:96
↓ 1 callersMethodlog_variant
(self, log_file, variant_data)
logger.py:416
↓ 1 callersMethodmerge_heads
(self, x)
decision_transformer/models/trajectory_gpt2.py:189
↓ 1 callersMethodpop_prefix
(self, )
logger.py:463
↓ 1 callersMethodprint_tabular
(self, new_tabular)
logger.py:215
↓ 1 callersMethodprune_heads
(self, heads)
decision_transformer/models/trajectory_gpt2.py:145
↓ 1 callersMethodrefresh
(self)
logger.py:223
↓ 1 callersFunctionsafe_json
(data)
logger.py:44
↓ 1 callersFunctionsave_checkpoint
(state,name)
experiment.py:43
↓ 1 callersMethodscale_up_eta
(self, lambda_)
decision_transformer/training/ql_trainer.py:162
↓ 1 callersMethodset_log_tabular_only
(self, log_tabular_only)
logger.py:339
↓ 1 callersFunctionset_seed
(seed)
experiment.py:55
↓ 1 callersMethodset_snapshot_dir
(self, dir_name)
logger.py:321
↓ 1 callersMethodset_snapshot_gap
(self, gap)
logger.py:336
↓ 1 callersMethodset_snapshot_mode
(self, mode)
logger.py:330
↓ 1 callersFunctionsetup_logger
Set up logger to have some reasonable default settings. Will save log output to based_log_dir/exp_prefix/exp_name. exp_name will
logger.py:96
↓ 1 callersMethodstep_ema
(self)
decision_transformer/training/ql_trainer.py:93
↓ 1 callersMethodtrain_iteration
(self, num_steps, logger, iter_num=0)
decision_transformer/training/trainer.py:22
↓ 1 callersMethodtrain_step
Train the model for one step states: (batch_size, max_len, state_dim)
decision_transformer/training/ql_trainer.py:165
↓ 1 callersMethodtrain_step
(self)
decision_transformer/training/trainer.py:65
↓ 1 callersMethodupdate_average
(self, old, new)
decision_transformer/training/ql_trainer.py:24
↓ 1 callersMethodupdate_model_average
(self, ma_model, current_model)
decision_transformer/training/ql_trainer.py:19
Method__init__
(self)
logger.py:211
Method__init__
(self, **kwargs)
experiment.py:39
Method__init__
(self, beta)
decision_transformer/training/ql_trainer.py:15
Method__init__
(self, model, critic, batch_size, tau,
decision_transformer/training/ql_trainer.py:32
Method__init__
(self, model, optimizer, batch_size, get_batch, loss_fn, scheduler=None, eval_fns=None)
decision_transformer/training/trainer.py:10
Method__init__
(self)
decision_transformer/envs/reacher_2d.py:11
Method__init__
( self, state_dim, act_dim, hidden_size, max_lengt
decision_transformer/models/decision_transformer.py:19
Method__init__
( self, state_dim, act_dim, hidden_size, max_lengt
decision_transformer/models/ql_DT.py:49
Method__init__
(self, nx, n_ctx, config, scale=False, is_cross_attention=False)
decision_transformer/models/trajectory_gpt2.py:121
Method__init__
(self, n_state, config)
decision_transformer/models/trajectory_gpt2.py:263
Method__init__
(self, n_ctx, config, scale=False)
decision_transformer/models/trajectory_gpt2.py:278
Method__init__
(self, *inputs, **kwargs)
decision_transformer/models/trajectory_gpt2.py:353
Method__init__
(self, config)
decision_transformer/models/trajectory_gpt2.py:517
Method__init__
(self, state_dim, act_dim, max_length=None)
decision_transformer/models/model.py:8
Function_column_type
The least generic type all column values are convertible to. >>> _column_type(["1", "2"]) is _int_type True >>> _column_type(["1", "2.3"]
tabulate.py:407
Method_init_weights
Initialize the weights.
decision_transformer/models/trajectory_gpt2.py:356
Function_latex_line_begin_tabular
(colwidths, colaligns)
tabulate.py:112
Function_mediawiki_row_with_attrs
(separator, cell_values, colwidths, colaligns)
tabulate.py:99
Function_more_generic
(type1, type2)
tabulate.py:400
Function_pipe_line_with_colons
Return a horizontal line with optional colons to indicate column's alignment (as in `pipe` output format).
tabulate.py:92
Method_prune_heads
Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
decision_transformer/models/trajectory_gpt2.py:578
Function_visible_width
Visible width of a printed string. ANSI color codes are removed. >>> _visible_width('\x1b[31mhello\x1b[0m'), _visible_width("world") (5, 5)
tabulate.py:349
Methodcreate_custom_forward
(module)
decision_transformer/models/trajectory_gpt2.py:716
Functioncreate_stats_ordered_dict
( name, data, stat_prefix=None, always_show_all_stats=True, exclude_ma
logger.py:163
Methodcustom_forward
(*inputs)
decision_transformer/models/trajectory_gpt2.py:717
Methoddefault
(self, o)
logger.py:233
Methoddeparallelize
(self)
decision_transformer/models/trajectory_gpt2.py:560
Functionevaluate_episode
( env, state_dim, act_dim, model, max_ep_len=1000, device='cud
decision_transformer/evaluation/evaluate_episodes.py:5
next →1–100 of 136, ranked by callers