MCPcopy Create free account

hub / github.com/SakanaAI/continuous-thought-machines / functions

Functions375 in github.com/SakanaAI/continuous-thought-machines

⨍ Functions 375◇ Types & classes 41
Functiongolden_test_expected_action_logits_rl
(device)
tests/conftest.py:226
Functiongolden_test_expected_action_probs_rl
(device)
tests/conftest.py:230
Functiongolden_test_expected_action_rl
(device)
tests/conftest.py:202
Functiongolden_test_expected_activated_state_trace_rl
(device)
tests/conftest.py:222
Functiongolden_test_expected_attentions_tracking_parity
()
tests/conftest.py:71
Functiongolden_test_expected_attentions_tracking_qamnist
()
tests/conftest.py:141
Functiongolden_test_expected_certainties_parity
(device)
tests/conftest.py:51
Functiongolden_test_expected_certainties_qamnist
(device)
tests/conftest.py:121
Functiongolden_test_expected_embeddings_tracking_qamnist
()
tests/conftest.py:145
Functiongolden_test_expected_initial_activated_state_trace_rl
(device)
tests/conftest.py:198
Functiongolden_test_expected_initial_state_trace_rl
(device)
tests/conftest.py:194
Functiongolden_test_expected_post_activations_tracking_parity
()
tests/conftest.py:67
Functiongolden_test_expected_post_activations_tracking_qamnist
()
tests/conftest.py:137
Functiongolden_test_expected_post_activations_tracking_rl
()
tests/conftest.py:238
Functiongolden_test_expected_pre_activations_tracking_parity
()
tests/conftest.py:63
Functiongolden_test_expected_pre_activations_tracking_qamnist
()
tests/conftest.py:133
Functiongolden_test_expected_pre_activations_tracking_rl
()
tests/conftest.py:234
Functiongolden_test_expected_predictions_parity
(device)
tests/conftest.py:47
Functiongolden_test_expected_predictions_qamnist
(device)
tests/conftest.py:117
Functiongolden_test_expected_state_trace_rl
(device)
tests/conftest.py:218
Functiongolden_test_expected_synch_out_tracking_rl
()
tests/conftest.py:242
Functiongolden_test_expected_synchronization_action_tracking_parity
()
tests/conftest.py:59
Functiongolden_test_expected_synchronization_action_tracking_qamnist
()
tests/conftest.py:129
Functiongolden_test_expected_synchronization_out_tracking_parity
()
tests/conftest.py:55
Functiongolden_test_expected_synchronization_out_tracking_qamnist
(device)
tests/conftest.py:125
Functiongolden_test_expected_value_rl
(device)
tests/conftest.py:214
Functiongolden_test_input_parity
(device)
tests/conftest.py:43
Functiongolden_test_input_qamnist
(device)
tests/conftest.py:91
Functiongolden_test_inputs_rl
(environment_to_test)
tests/conftest.py:187
Functiongolden_test_model_rl
(golden_test_params_rl, device)
tests/conftest.py:161
Functiongolden_test_params_parity
(parity_params)
tests/conftest.py:31
Functiongolden_test_params_qamnist
(base_params)
tests/conftest.py:77
Functiongolden_test_params_rl
(base_params)
tests/conftest.py:151
Functionload_training_data_from_checkpoint
Loads only training data (losses and accuracies) from the checkpoint without loading the model.
tasks/qamnist/analysis/run.py:30
Functionmake_parity_gif
(predictions, certainties, targets, pre_activations, post_activations, attention_weights, inputs_to_model, fil
tasks/parity/plotting.py:24
Functionmake_qamnist_gif
(predictions, certainties, targets, pre_activations, post_activations, input_gates, inputs_to_model, filename,
tasks/qamnist/plotting.py:14
Functionmake_rl_gif
(action_logits, action_probs, actions, values, rewards, pre_activations, post_activations, inputs, filename)
tasks/rl/plotting.py:11
Functionparity_input
(device)
tests/conftest.py:271
Functionparity_params
(base_params)
tests/conftest.py:285
Functionparse_env_id
(env_id)
tasks/rl/analysis/run.py:229
Functionplot_loss_all_runs
(training_data, evaluate_every, save_path="train_loss_comparison_qamnist.png", window_size=1000)
tasks/qamnist/analysis/run.py:110
Functionqamnist_input
(device)
tests/conftest.py:325
Functionqamnist_params
()
tests/conftest.py:301
Functionresnet101
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True,
models/resnet.py:331
Functionresnet152
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True,
models/resnet.py:342
Functionresnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True,
models/resnet.py:298
Functionresnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True,
models/resnet.py:309
Functionresnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet progress (bool): If True,
models/resnet.py:320
Functionseed
()
tests/conftest.py:19
Methodset_backbone
Set the backbone module based on the specified type.
models/ctm_rl.py:84
Methodset_backbone
Set the backbone module based on the specified type.
models/ctm_qamnist.py:64
Methodset_initial_rgb
Set the initial RGB values for the backbone.
models/ctm_rl.py:76
Methodset_initial_rgb
Set the initial RGB values for the backbone.
models/ctm_qamnist.py:56
Methodset_synchronisation_parameters
Set the parameters for the synchronisation of neurons.
models/ctm_rl.py:121
Methodstep
(self, epoch=None)
utils/schedulers.py:91
Functiontest_golden_parity
Golden test the parity CTM model.
tests/tests.py:30
Functiontest_golden_qamnist
Golden test the QAMNIST CTM model.
tests/tests.py:48
Functiontest_golden_rl
(golden_test_model_rl, golden_test_inputs_rl, golden_test_expected_initial_state_trace_rl, golden_test_expecte
tests/tests.py:68
Functiontest_invalid_neuron_select_type
(ctm_factory, base_params)
tests/tests.py:152
Functiontest_none_backbone_with_none_positional_embeddings
(ctm_factory, base_params)
tests/tests.py:167
Functiontest_none_backbone_with_valid_positional_embeddings
(ctm_factory, base_params, positional_embedding_type)
tests/tests.py:176
Functiontest_none_neuron_select_type
(ctm_factory, base_params)
tests/tests.py:148
Functiontest_parity_certainty_shape
(parity_ctm_model, parity_params, parity_input)
tests/tests.py:202
Functiontest_parity_nans_in_predictions
(parity_ctm_model, parity_input)
tests/tests.py:209
Functiontest_parity_prediction_shape
(parity_ctm_model, parity_params, parity_input)
tests/tests.py:195
Functiontest_qamnist_certainty_shape
(qamnist_model_factory, qamnist_params, qamnist_input, device)
tests/tests.py:226
Functiontest_qamnist_nans_in_predictions
(qamnist_model_factory, qamnist_input, device)
tests/tests.py:236
Functiontest_qamnist_prediction_shape
(qamnist_model_factory, qamnist_params, qamnist_input, device)
tests/tests.py:215
Functiontest_qamnist_synchronisation_shape
(qamnist_model_factory, qamnist_params, qamnist_input, neuron_select_type, device)
tests/tests.py:244
Functiontest_set_synchronisation_parameters
(ctm_factory, base_params, device, synch_type, neuron_select_type)
tests/tests.py:103
Functiontest_valid_backbone_and_valid_positional_embedding
(ctm_factory, base_params, backbone_type, positional_embedding_type)
tests/tests.py:159
Functiontest_valid_backbone_with_none_positional_embeddings
(ctm_factory, base_params, backbone_type)
tests/tests.py:185
Functiontest_valid_neuron_select_type
(ctm_factory, base_params, neuron_select_type)
tests/tests.py:144
Functionthunk
()
tasks/rl/train.py:86
Methodverify_args
Verify the validity of the input arguments.
models/ctm_rl.py:136
← previous301–375 of 375, ranked by callers