MCPcopy Create free account

hub / github.com/charleshsc/QT / functions

Functions136 in github.com/charleshsc/QT

⨍ Functions 136◇ Types & classes 21↳ Endpoints 1
Functionfn
(model, critic)
experiment.py:299
Methodforward
( self, hidden_states, layer_past=None, attention_mask=None,
decision_transformer/models/trajectory_gpt2.py:202
Methodforward
(self, x)
decision_transformer/models/trajectory_gpt2.py:256
Methodforward
(self, x)
decision_transformer/models/trajectory_gpt2.py:271
Methodforward
( self, hidden_states, layer_past=None, attention_mask=None,
decision_transformer/models/trajectory_gpt2.py:292
Methodforward
( self, input_ids=None, past_key_values=None, attention_mask=N
decision_transformer/models/trajectory_gpt2.py:592
Methodforward
(self, states, actions, rewards, masks=None, attention_mask=None)
decision_transformer/models/model.py:15
Methodget_action
(self, states, actions, rewards=None, returns_to_go=None, timesteps=None, **kwargs)
decision_transformer/models/decision_transformer.py:155
Methodget_action
(self, critic, states, actions, rewards=None, returns_to_go=None, timesteps=None, **kwargs)
decision_transformer/models/ql_DT.py:158
Functionget_batch
(batch_size=256, max_len=K)
experiment.py:239
Methodget_input_embeddings
(self)
decision_transformer/models/trajectory_gpt2.py:572
Methodget_log_tabular_only
(self, )
logger.py:342
Methodget_snapshot_dir
(self, )
logger.py:324
Methodget_snapshot_gap
(self, )
logger.py:333
Methodget_snapshot_mode
(self, )
logger.py:327
Methodget_table_dict
(self, )
logger.py:396
Methodget_table_key_set
(self, )
logger.py:399
Functionload_tf_weights_in_gpt2
Load tf checkpoints in a pytorch model
decision_transformer/models/trajectory_gpt2.py:63
Methodparallelize
(self, device_map=None)
decision_transformer/models/trajectory_gpt2.py:540
Methodprefix
(self, key)
logger.py:403
Methodq1
(self, state, action)
decision_transformer/models/ql_DT.py:34
Methodrecord_dict
(self, d, prefix=None)
logger.py:364
Methodrecord_tabular_misc_stat
(self, key, values, placement='back')
logger.py:421
Methodremove_tabular_output
(self, file_name, relative_to_snapshot_dir=False)
logger.py:314
Methodremove_text_output
(self, file_name)
logger.py:305
Methodreset_model
(self)
decision_transformer/envs/reacher_2d.py:50
Methodsave_extra_data
Data saved here will always override the last entry :param data: Something pickle'able.
logger.py:380
Methodsave_itr_params
(self, itr, params)
logger.py:467
Methodset_input_embeddings
(self, new_embeddings)
decision_transformer/models/trajectory_gpt2.py:575
Methodset_layers
(self, num_layers)
decision_transformer/models/trajectory_gpt2.py:533
Functionsimple_separated_format
Construct a simple TableFormat with columns separated by a separator. >>> tsv = simple_separated_format("\\t") ; \ tabulate([["foo", 1],
tabulate.py:198
Methodtabular_prefix
(self, key)
logger.py:411
Methodtrain_iteration
(self, num_steps, logger, iter_num=0, log_writer=None)
decision_transformer/training/ql_trainer.py:97
Methodtrain_step
(self)
decision_transformer/training/act_trainer.py:9
Methodtrain_step
(self)
decision_transformer/training/seq_trainer.py:9
Methodviewer_setup
(self)
decision_transformer/envs/reacher_2d.py:64
← previous101–136 of 136, ranked by callers