Index

_

_apply_meta_update() (ReferentialGym.modules.population_handler_module.PopulationHandlerModule method) _check_exists() (ReferentialGym.datasets.CLEVR_dataset.CLEVRDataset method) (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) _compute_feat_map() (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.networks.EntityPrioredConvolutionalBody method) (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) _compute_feature_shape() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) (ReferentialGym.networks.residual_networks.ResNet18AvgPooledMHDPA method) _compute_features() (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) _compute_sentence() (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) _compute_tau() (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.caption_speaker.CaptionSpeaker method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.discriminative_listener.DiscriminativeListener method) (ReferentialGym.agents.eos_priored_lstm_cnn_speaker.EoSPrioredLSTMCNNSpeaker method) (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.lstm_cnn_speaker.LSTMCNNSpeaker method) (ReferentialGym.agents.multi_head_lstm_cnn_speaker.MultiHeadLSTMCNNSpeaker method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.speaker.Speaker method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method) (ReferentialGym.agents.transcoding_lstm_cnn_speaker.TranscodingLSTMCNNSpeaker method) _download() (ReferentialGym.datasets.CLEVR_dataset.CLEVRDataset method) _forward() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) _forward_impl() (ReferentialGym.networks.residual_networks.CoordResNet method) (ReferentialGym.networks.residual_networks.ResNet method) _generate() (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) _generate_training_batch() (ReferentialGym.modules.factor_vae_disentanglement_metric_module.FactorVAEDisentanglementMetricModule method) _generate_training_sample() (ReferentialGym.modules.factor_vae_disentanglement_metric_module.FactorVAEDisentanglementMetricModule method) _initialize_weights() (ReferentialGym.networks.networks.VGG method) _log() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) _make_layer() (ReferentialGym.networks.residual_networks.CoordResNet method) (ReferentialGym.networks.residual_networks.ResNet method) _makeSymbolicXYTSfeatures() (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) _makeVisualXYTSfeatures() (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) _operate() (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method) _prune_dims() (ReferentialGym.modules.factor_vae_disentanglement_metric_module.FactorVAEDisentanglementMetricModule method) _reason() (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.discriminative_listener.DiscriminativeListener method) (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method)

_reptile_step() (ReferentialGym.modules.population_handler_module.PopulationHandlerModule method) _reset_rnn_states() (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.speaker.Speaker method) _select_agents() (ReferentialGym.modules.population_handler_module.PopulationHandlerModule method) _sense() (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.caption_speaker.CaptionSpeaker method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.discriminative_listener.DiscriminativeListener method) (ReferentialGym.agents.eos_priored_lstm_cnn_speaker.EoSPrioredLSTMCNNSpeaker method) (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.lstm_cnn_speaker.LSTMCNNSpeaker method) (ReferentialGym.agents.multi_head_lstm_cnn_speaker.MultiHeadLSTMCNNSpeaker method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.speaker.Speaker method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method) (ReferentialGym.agents.transcoding_lstm_cnn_speaker.TranscodingLSTMCNNSpeaker method) _serve_module() (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method) _tidyup() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.eos_priored_lstm_cnn_speaker.EoSPrioredLSTMCNNSpeaker method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.lstm_cnn_speaker.LSTMCNNSpeaker method) (ReferentialGym.agents.multi_head_lstm_cnn_speaker.MultiHeadLSTMCNNSpeaker method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method) (ReferentialGym.agents.transcoding_lstm_cnn_speaker.TranscodingLSTMCNNSpeaker method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) _utter() (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.caption_speaker.CaptionSpeaker method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.discriminative_listener.DiscriminativeListener method) (ReferentialGym.agents.eos_priored_lstm_cnn_speaker.EoSPrioredLSTMCNNSpeaker method) (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.lstm_cnn_speaker.LSTMCNNSpeaker method) (ReferentialGym.agents.multi_head_lstm_cnn_speaker.MultiHeadLSTMCNNSpeaker method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.speaker.Speaker method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method) (ReferentialGym.agents.transcoding_lstm_cnn_speaker.TranscodingLSTMCNNSpeaker method) _vgg() (in module ReferentialGym.networks.networks)

A

add() (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) add_dict() (ReferentialGym.utils.statsLogger.statsLogger.statsLogger method) add_key() (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) AddEgocentricInvariance (class in ReferentialGym.datasets.utils) addXYfeatures (class in ReferentialGym.networks.networks)

addXYRhoThetaFeatures (class in ReferentialGym.networks.networks) addXYSfeatures (class in ReferentialGym.networks.autoregressive_networks) Agent (class in ReferentialGym.agents.agent) AttentionLSTMCNNListener (class in ReferentialGym.agents.attention_lstm_cnn_listener) AttentionNetwork (class in ReferentialGym.networks.autoregressive_networks)

B

BatchReshapeModule (class in ReferentialGym.modules.batch_reshape_module) BatchReshapeRepeatModule (class in ReferentialGym.modules.batch_reshape_repeat_module) Bernoulli (class in ReferentialGym.networks.autoregressive_networks) BetaVAE (class in ReferentialGym.networks.autoregressive_networks) bookkeeping() (ReferentialGym.modules.population_handler_module.PopulationHandlerModule method) BroadcastingDecoder (class in ReferentialGym.networks.autoregressive_networks) BroadcastingDeconvDecoder (class in ReferentialGym.networks.autoregressive_networks) build_BatchReshapeModule() (in module ReferentialGym.modules.batch_reshape_module) build_BatchReshapeRepeatModule() (in module ReferentialGym.modules.batch_reshape_repeat_module) build_ConcatModule() (in module ReferentialGym.modules.concat_module) build_dSpritesPerLatentAccuracyMetricModule() (in module ReferentialGym.modules.dsprites_per_latent_accuracy_metric_module) build_FactorVAEDisentanglementMetricModule() (in module ReferentialGym.modules.factor_vae_disentanglement_metric_module)

build_FlattenModule() (in module ReferentialGym.modules.flatten_module) build_GradRecorderModule() (in module ReferentialGym.modules.grad_recorder_module) build_HomoscedasticMultiTasksLossModule() (in module ReferentialGym.modules.homoscedastic_multi_task_loss_module) build_InstantaneousCoordinationMetricModule() (in module ReferentialGym.modules.instantaneous_coordination_metric_module) build_MultiHeadClassificationFromFeatureMapModule() (in module ReferentialGym.modules.multi_head_classification_from_feature_map_module) build_MultiHeadClassificationModule() (in module ReferentialGym.modules.multi_head_classification_module) build_MultiHeadRegressionModule() (in module ReferentialGym.modules.multi_head_regression_module) build_OptimizationModule() (in module ReferentialGym.modules.optimization_module) build_PerEpochLoggerModule() (in module ReferentialGym.modules.per_epoch_logger_module) build_PopulationHandlerModule() (in module ReferentialGym.modules.population_handler_module) build_SqueezeModule() (in module ReferentialGym.modules.squeeze_module) build_TopographicSimilarityMetricModule() (in module ReferentialGym.modules.topographic_similarity_metric_module) build_VisualModule() (in module ReferentialGym.modules.visual_module)

C

CaptionSpeaker (class in ReferentialGym.agents.caption_speaker) cardinality() (in module ReferentialGym.utils.utils) cat() (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) CategoricalObverterAgent (class in ReferentialGym.agents.categorical_obverter_agent) choose_architecture() (in module ReferentialGym.networks) CLEVRDataset (class in ReferentialGym.datasets.CLEVR_dataset) clone() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) collate_dict_wrapper() (in module ReferentialGym.datasets.utils) compute() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.batch_reshape_module.BatchReshapeModule method) (ReferentialGym.modules.batch_reshape_repeat_module.BatchReshapeRepeatModule method) (ReferentialGym.modules.concat_module.ConcatModule method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) (ReferentialGym.modules.dsprites_per_latent_accuracy_metric_module.dSpritesPerLatentAccuracyMetricModule method) (ReferentialGym.modules.factor_vae_disentanglement_metric_module.FactorVAEDisentanglementMetricModule method) (ReferentialGym.modules.flatten_module.FlattenModule method) (ReferentialGym.modules.grad_recorder_module.GradRecorderModule method) (ReferentialGym.modules.homoscedastic_multi_task_loss_module.HomoscedasticMultiTasksLossModule method) (ReferentialGym.modules.instantaneous_coordination_metric_module.InstantaneousCoordinationMetricModule method) (ReferentialGym.modules.module.Module method) (ReferentialGym.modules.multi_head_classification_from_feature_map_module.MultiHeadClassificationFromFeatureMapModule method) (ReferentialGym.modules.multi_head_classification_module.MultiHeadClassificationModule method) (ReferentialGym.modules.multi_head_regression_module.MultiHeadRegressionModule method) (ReferentialGym.modules.optimization_module.OptimizationModule method) (ReferentialGym.modules.per_epoch_logger_module.PerEpochLoggerModule method) (ReferentialGym.modules.population_handler_module.PopulationHandlerModule method) (ReferentialGym.modules.squeeze_module.SqueezeModule method) (ReferentialGym.modules.topographic_similarity_metric_module.TopographicSimilarityMetricModule method) (ReferentialGym.modules.visual_module.VisualModule method)

compute_cosine_sim() (in module ReferentialGym.utils.utils) compute_cosine_sim_for_idx_over_comprange() (in module ReferentialGym.utils.utils) compute_levenshtein_distance() (in module ReferentialGym.utils.utils) compute_levenshtein_distance_for_idx_over_comprange() (in module ReferentialGym.utils.utils) compute_loss() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.MONet method) (ReferentialGym.networks.autoregressive_networks.ParallelMONet method) compute_reinforce_losses() (in module ReferentialGym.agents.discriminative_listener) compute_topographic_similarity_parallel() (in module ReferentialGym.utils.utils) ConcatModule (class in ReferentialGym.modules.concat_module) conv() (in module ReferentialGym.networks.networks) conv1x1() (in module ReferentialGym.networks.networks) conv3x3() (in module ReferentialGym.networks.networks) ConvolutionalBody (class in ReferentialGym.networks.networks) ConvolutionalGruBody (class in ReferentialGym.networks.networks) ConvolutionalLstmBody (class in ReferentialGym.networks.networks) ConvolutionalMHDPABody (class in ReferentialGym.networks.networks) coord4conv() (in module ReferentialGym.networks.networks) coord4conv1x1() (in module ReferentialGym.networks.networks) coord4conv3x3() (in module ReferentialGym.networks.networks) coord4deconv() (in module ReferentialGym.networks.networks) coordconv() (in module ReferentialGym.networks.networks) coordconv1x1() (in module ReferentialGym.networks.networks) coordconv3x3() (in module ReferentialGym.networks.networks) coorddeconv() (in module ReferentialGym.networks.networks) CoordResNet (class in ReferentialGym.networks.residual_networks) counter_test_threshold (ReferentialGym.datasets.dSprites_dataset.dSpritesDataset attribute) cuda() (ReferentialGym.datasets.utils.DictBatch method) CurrentAgentModule (class in ReferentialGym.modules.current_agent_module)

D

Dataset (class in ReferentialGym.datasets.dataset) decode() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.BroadcastingDecoder method) (ReferentialGym.networks.autoregressive_networks.BroadcastingDeconvDecoder method) (ReferentialGym.networks.autoregressive_networks.Decoder method) (ReferentialGym.networks.autoregressive_networks.MONet method) (ReferentialGym.networks.autoregressive_networks.ParallelAttentionBroadcastingDeconvDecoder method) (ReferentialGym.networks.autoregressive_networks.ParallelMONet method) Decoder (class in ReferentialGym.networks.autoregressive_networks) deconv() (in module ReferentialGym.networks.networks) DictBatch (class in ReferentialGym.datasets.utils)

DictDatasetWrapper (class in ReferentialGym.datasets.dict_dataset_wrapper) DifferentiableObverterAgent (class in ReferentialGym.agents.differentiable_obverter_agent) DifferentiableRelationalObverterAgent (class in ReferentialGym.agents.differentiable_relational_obverter) discriminative_reinforce_referential_game_loss() (in module ReferentialGym.agents.discriminative_listener) discriminative_st_gs_referential_game_loss() (in module ReferentialGym.agents.discriminative_listener) DiscriminativeListener (class in ReferentialGym.agents.discriminative_listener) Distribution (class in ReferentialGym.networks.autoregressive_networks) download_preprocess_dataset() (in module ReferentialGym.datasets.CLEVR_dataset) dSpritesDataset (class in ReferentialGym.datasets.dSprites_dataset) dSpritesPerLatentAccuracyMetricModule (class in ReferentialGym.modules.dsprites_per_latent_accuracy_metric_module) DualLabeledDataset (class in ReferentialGym.datasets.dual_labeled_dataset) dump() (ReferentialGym.utils.statsLogger.statsLogger.statsLogger method)

E

encode() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.ResNetAvgPooledEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetParallelAttentionEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetPHDPAEncoder method) encodeZ() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.MONet method) (ReferentialGym.networks.autoregressive_networks.ParallelMONet method)

EntityPrioredConvolutionalBody (class in ReferentialGym.networks.networks) entropy_logging_hook() (in module ReferentialGym.agents.speaker) entropy_regularization_loss_hook() (in module ReferentialGym.agents.speaker) eos_priored_loss_hook() (in module ReferentialGym.agents.eos_priored_lstm_cnn_speaker) EoSPrioredLSTMCNNSpeaker (class in ReferentialGym.agents.eos_priored_lstm_cnn_speaker) ExperienceBuffer (class in ReferentialGym.agents.discriminative_listener) ExtractorResNet18 (class in ReferentialGym.networks.residual_networks) ExtractorVGG16 (class in ReferentialGym.networks.networks)

F

FactorVAEDisentanglementMetricModule (class in ReferentialGym.modules.factor_vae_disentanglement_metric_module) FCBody (class in ReferentialGym.networks.networks) FlattenModule (class in ReferentialGym.modules.flatten_module) forward() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.agents.discriminative_listener.DiscriminativeListener method) (ReferentialGym.agents.generative_listener.GenerativeListener method) (ReferentialGym.agents.speaker.Speaker method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) (ReferentialGym.networks.autoregressive_networks.addXYSfeatures method) (ReferentialGym.networks.autoregressive_networks.AttentionNetwork method) (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.BroadcastingDecoder method) (ReferentialGym.networks.autoregressive_networks.BroadcastingDeconvDecoder method) (ReferentialGym.networks.autoregressive_networks.Decoder method) (ReferentialGym.networks.autoregressive_networks.MONet method) (ReferentialGym.networks.autoregressive_networks.ParallelAttentionBroadcastingDeconvDecoder method) (ReferentialGym.networks.autoregressive_networks.ParallelAttentionNetwork method) (ReferentialGym.networks.autoregressive_networks.ParallelMONet method) (ReferentialGym.networks.autoregressive_networks.ResNetAvgPooledEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetParallelAttentionEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetPHDPAEncoder method) (ReferentialGym.networks.autoregressive_networks.UNet method) (ReferentialGym.networks.autoregressive_networks.UNetBlock method) (ReferentialGym.networks.homoscedastic_multitask_loss.HomoscedasticMultiTasksLoss method) (ReferentialGym.networks.networks.addXYfeatures method) (ReferentialGym.networks.networks.addXYRhoThetaFeatures method) (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.networks.ConvolutionalGruBody method) (ReferentialGym.networks.networks.ConvolutionalLstmBody method) (ReferentialGym.networks.networks.ConvolutionalMHDPABody method) (ReferentialGym.networks.networks.ExtractorVGG16 method) (ReferentialGym.networks.networks.FCBody method) (ReferentialGym.networks.networks.GRUBody method) (ReferentialGym.networks.networks.LSTMBody method) (ReferentialGym.networks.networks.MHDPA method) (ReferentialGym.networks.networks.MHDPA_RN method) (ReferentialGym.networks.networks.ModelVGG16 method) (ReferentialGym.networks.networks.VGG method) (ReferentialGym.networks.residual_networks.CoordResNet method) (ReferentialGym.networks.residual_networks.ExtractorResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) (ReferentialGym.networks.residual_networks.ResNet method) (ReferentialGym.networks.residual_networks.ResNet18AvgPooledMHDPA method) (ReferentialGym.networks.residual_networks.ResNet18MHDPA method) (ReferentialGym.utils.utils.StraightThroughGumbelSoftmaxLayer method)

forwardScaledDPAhead() (ReferentialGym.networks.networks.MHDPA_RN method) forwardStackedMHDPA() (ReferentialGym.networks.networks.MHDPA_RN method)

G

generate_dataset() (in module ReferentialGym.datasets.extended_sort_of_CLEVR_dataset) (in module ReferentialGym.datasets.sort_of_CLEVR_dataset) generative_st_gs_referential_game_loss() (in module ReferentialGym.agents.generative_listener) GenerativeListener (class in ReferentialGym.agents.generative_listener) get_feat_map() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) get_feature_shape() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) (ReferentialGym.networks.autoregressive_networks.MONet method) (ReferentialGym.networks.autoregressive_networks.ParallelMONet method) (ReferentialGym.networks.autoregressive_networks.ResNetAvgPooledEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetParallelAttentionEncoder method) (ReferentialGym.networks.autoregressive_networks.ResNetPHDPAEncoder method) (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.networks.ConvolutionalGruBody method) (ReferentialGym.networks.networks.ConvolutionalLstmBody method) (ReferentialGym.networks.networks.FCBody method) (ReferentialGym.networks.networks.GRUBody method) (ReferentialGym.networks.networks.LSTMBody method) (ReferentialGym.networks.networks.ModelVGG16 method) (ReferentialGym.networks.residual_networks.ModelResNet18 method) (ReferentialGym.networks.residual_networks.ModelResNet18AvgPooled method) get_id() (ReferentialGym.modules.module.Module method) get_input_shape() (ReferentialGym.networks.networks.ConvolutionalBody method) (ReferentialGym.networks.networks.ConvolutionalGruBody method) (ReferentialGym.networks.networks.ConvolutionalLstmBody method) get_input_stream_ids() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) (ReferentialGym.modules.module.Module method)

get_input_stream_keys() (ReferentialGym.agents.agent.Agent method) get_reset_states() (ReferentialGym.networks.networks.ConvolutionalGruBody method) (ReferentialGym.networks.networks.ConvolutionalLstmBody method) (ReferentialGym.networks.networks.GRUBody method) (ReferentialGym.networks.networks.LSTMBody method) get_type() (ReferentialGym.modules.module.Module method) getAnswerVocabSize() (ReferentialGym.datasets.CLEVR_dataset.CLEVRDataset method) getclass() (ReferentialGym.datasets.CLEVR_dataset.CLEVRDataset method) (ReferentialGym.datasets.dict_dataset_wrapper.DictDatasetWrapper method) (ReferentialGym.datasets.dSprites_dataset.dSpritesDataset method) (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.MSCOCO_dataset.MSCOCODataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) getlatentclass() (ReferentialGym.datasets.dSprites_dataset.dSpritesDataset method) (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) getlatentonehot() (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) getlatentvalue() (ReferentialGym.datasets.dSprites_dataset.dSpritesDataset method) (ReferentialGym.datasets.extended_sort_of_CLEVR_dataset.XSortOfCLEVRDataset method) (ReferentialGym.datasets.MSCOCO_dataset.MSCOCODataset method) (ReferentialGym.datasets.sort_of_CLEVR_dataset.SortOfCLEVRDataset method) getNbrClasses() (ReferentialGym.datasets.dataset.Dataset method) (ReferentialGym.datasets.dual_labeled_dataset.DualLabeledDataset method) (ReferentialGym.datasets.labeled_dataset.LabeledDataset method) getNbrDistractors() (ReferentialGym.datasets.dataset.Dataset method) gettestlatentmask() (ReferentialGym.datasets.dSprites_dataset.dSpritesDataset method) getVocabSize() (ReferentialGym.datasets.CLEVR_dataset.CLEVRDataset method) GradRecorderModule (class in ReferentialGym.modules.grad_recorder_module) GRUBody (class in ReferentialGym.networks.networks) gumbel_softmax() (in module ReferentialGym.utils.utils)

H

handle_nan() (in module ReferentialGym.networks.networks) hasnan() (in module ReferentialGym.networks.networks) havrylov_hinge_learning_signal() (in module ReferentialGym.agents.discriminative_listener) (in module ReferentialGym.agents.generative_listener)

HomoscedasticMultiTasksLoss (class in ReferentialGym.networks.homoscedastic_multitask_loss) HomoscedasticMultiTasksLossModule (class in ReferentialGym.modules.homoscedastic_multi_task_loss_module)

I

in_channel (ReferentialGym.networks.autoregressive_networks.AttentionNetwork attribute)

InstantaneousCoordinationMetricModule (class in ReferentialGym.modules.instantaneous_coordination_metric_module)

K

keys() (ReferentialGym.datasets.utils.DictBatch method)

L

LabeledDataset (class in ReferentialGym.datasets.labeled_dataset) layer_init() (in module ReferentialGym.networks.networks) Listener (class in ReferentialGym.agents.listener) load() (ReferentialGym.networks.networks.MHDPA method) log_prob() (ReferentialGym.networks.autoregressive_networks.Bernoulli method) (ReferentialGym.networks.autoregressive_networks.Distribution method) (ReferentialGym.networks.autoregressive_networks.Normal method)

LSTMBody (class in ReferentialGym.networks.networks) LSTMCNNListener (class in ReferentialGym.agents.lstm_cnn_listener) LSTMCNNSpeaker (class in ReferentialGym.agents.lstm_cnn_speaker)

M

make() (in module ReferentialGym) make_tl_dataset() (ReferentialGym.datasets.MSCOCO_dataset.MSCOCODataset method) maxl1_loss_hook() (in module ReferentialGym.agents.agent) mdl_principle_loss_hook() (in module ReferentialGym.agents.speaker) measure_topographic_similarity() (ReferentialGym.utils.statsLogger.statsLogger.statsLogger method) MHDPA (class in ReferentialGym.networks.networks) MHDPA_RN (class in ReferentialGym.networks.networks) ModelResNet18 (class in ReferentialGym.networks.residual_networks) ModelResNet18AvgPooled (class in ReferentialGym.networks.residual_networks) ModelVGG16 (class in ReferentialGym.networks.networks) module ReferentialGym ReferentialGym.agents ReferentialGym.agents.agent ReferentialGym.agents.attention_lstm_cnn_listener ReferentialGym.agents.caption_speaker ReferentialGym.agents.categorical_obverter_agent ReferentialGym.agents.differentiable_obverter_agent ReferentialGym.agents.differentiable_relational_obverter ReferentialGym.agents.discriminative_listener ReferentialGym.agents.eos_priored_lstm_cnn_speaker ReferentialGym.agents.generative_listener ReferentialGym.agents.listener ReferentialGym.agents.lstm_cnn_listener ReferentialGym.agents.lstm_cnn_speaker ReferentialGym.agents.multi_head_lstm_cnn_speaker ReferentialGym.agents.obverter_agent ReferentialGym.agents.speaker ReferentialGym.agents.transcoding_lstm_cnn_listener ReferentialGym.agents.transcoding_lstm_cnn_speaker ReferentialGym.datasets ReferentialGym.datasets.CLEVR_dataset ReferentialGym.datasets.dataset ReferentialGym.datasets.dict_dataset_wrapper ReferentialGym.datasets.dSprites_dataset ReferentialGym.datasets.dual_labeled_dataset ReferentialGym.datasets.extended_sort_of_CLEVR_dataset ReferentialGym.datasets.labeled_dataset ReferentialGym.datasets.MSCOCO_dataset ReferentialGym.datasets.sort_of_CLEVR_dataset ReferentialGym.datasets.utils ReferentialGym.modules ReferentialGym.modules.batch_reshape_module ReferentialGym.modules.batch_reshape_repeat_module ReferentialGym.modules.concat_module ReferentialGym.modules.current_agent_module ReferentialGym.modules.dsprites_per_latent_accuracy_metric_module ReferentialGym.modules.factor_vae_disentanglement_metric_module ReferentialGym.modules.flatten_module ReferentialGym.modules.grad_recorder_module ReferentialGym.modules.homoscedastic_multi_task_loss_module ReferentialGym.modules.instantaneous_coordination_metric_module ReferentialGym.modules.module ReferentialGym.modules.multi_head_classification_from_feature_map_module ReferentialGym.modules.multi_head_classification_module ReferentialGym.modules.multi_head_regression_module ReferentialGym.modules.optimization_module ReferentialGym.modules.per_epoch_logger_module ReferentialGym.modules.population_handler_module ReferentialGym.modules.squeeze_module ReferentialGym.modules.topographic_similarity_metric_module ReferentialGym.modules.visual_module ReferentialGym.networks ReferentialGym.networks.autoregressive_networks ReferentialGym.networks.homoscedastic_multitask_loss ReferentialGym.networks.networks ReferentialGym.networks.residual_networks ReferentialGym.referential_game ReferentialGym.utils ReferentialGym.utils.statsLogger ReferentialGym.utils.statsLogger.statsLogger ReferentialGym.utils.streamHandler ReferentialGym.utils.streamHandler.StreamHandler ReferentialGym.utils.utils

Module (class in ReferentialGym.modules.module) MONet (class in ReferentialGym.networks.autoregressive_networks) MSCOCODataset (class in ReferentialGym.datasets.MSCOCO_dataset) MultiHeadClassificationFromFeatureMapModule (class in ReferentialGym.modules.multi_head_classification_from_feature_map_module) MultiHeadClassificationModule (class in ReferentialGym.modules.multi_head_classification_module) MultiHeadLSTMCNNSpeaker (class in ReferentialGym.agents.multi_head_lstm_cnn_speaker) MultiHeadRegressionModule (class in ReferentialGym.modules.multi_head_regression_module)

N

Normal (class in ReferentialGym.networks.autoregressive_networks)

O

ObverterAgent (class in ReferentialGym.agents.obverter_agent)

oov_loss_hook() (in module ReferentialGym.agents.speaker) OptimizationModule (class in ReferentialGym.modules.optimization_module)

P

ParallelAttentionBroadcastingDeconvDecoder (class in ReferentialGym.networks.autoregressive_networks) ParallelAttentionNetwork (class in ReferentialGym.networks.autoregressive_networks) ParallelMONet (class in ReferentialGym.networks.autoregressive_networks) penalize_multi_round_binary_reward_fn() (in module ReferentialGym.agents.discriminative_listener)

PerEpochLoggerModule (class in ReferentialGym.modules.per_epoch_logger_module) permutate_latents() (ReferentialGym.networks.autoregressive_networks.TotalCorrelationDiscriminator method) pin_memory() (ReferentialGym.datasets.utils.DictBatch method) pop() (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) PopulationHandlerModule (class in ReferentialGym.modules.population_handler_module)

Q

query_vae_latent_space() (in module ReferentialGym.utils.utils)

R

ReferentialGame (class in ReferentialGym.referential_game) ReferentialGym module ReferentialGym.agents module ReferentialGym.agents.agent module ReferentialGym.agents.attention_lstm_cnn_listener module ReferentialGym.agents.caption_speaker module ReferentialGym.agents.categorical_obverter_agent module ReferentialGym.agents.differentiable_obverter_agent module ReferentialGym.agents.differentiable_relational_obverter module ReferentialGym.agents.discriminative_listener module ReferentialGym.agents.eos_priored_lstm_cnn_speaker module ReferentialGym.agents.generative_listener module ReferentialGym.agents.listener module ReferentialGym.agents.lstm_cnn_listener module ReferentialGym.agents.lstm_cnn_speaker module ReferentialGym.agents.multi_head_lstm_cnn_speaker module ReferentialGym.agents.obverter_agent module ReferentialGym.agents.speaker module ReferentialGym.agents.transcoding_lstm_cnn_listener module ReferentialGym.agents.transcoding_lstm_cnn_speaker module ReferentialGym.datasets module ReferentialGym.datasets.CLEVR_dataset module ReferentialGym.datasets.dataset module ReferentialGym.datasets.dict_dataset_wrapper module ReferentialGym.datasets.dSprites_dataset module ReferentialGym.datasets.dual_labeled_dataset module ReferentialGym.datasets.extended_sort_of_CLEVR_dataset module ReferentialGym.datasets.labeled_dataset module ReferentialGym.datasets.MSCOCO_dataset module ReferentialGym.datasets.sort_of_CLEVR_dataset module ReferentialGym.datasets.utils module ReferentialGym.modules module ReferentialGym.modules.batch_reshape_module module ReferentialGym.modules.batch_reshape_repeat_module module ReferentialGym.modules.concat_module module ReferentialGym.modules.current_agent_module module ReferentialGym.modules.dsprites_per_latent_accuracy_metric_module module ReferentialGym.modules.factor_vae_disentanglement_metric_module module ReferentialGym.modules.flatten_module module ReferentialGym.modules.grad_recorder_module module

ReferentialGym.modules.homoscedastic_multi_task_loss_module module ReferentialGym.modules.instantaneous_coordination_metric_module module ReferentialGym.modules.module module ReferentialGym.modules.multi_head_classification_from_feature_map_module module ReferentialGym.modules.multi_head_classification_module module ReferentialGym.modules.multi_head_regression_module module ReferentialGym.modules.optimization_module module ReferentialGym.modules.per_epoch_logger_module module ReferentialGym.modules.population_handler_module module ReferentialGym.modules.squeeze_module module ReferentialGym.modules.topographic_similarity_metric_module module ReferentialGym.modules.visual_module module ReferentialGym.networks module ReferentialGym.networks.autoregressive_networks module ReferentialGym.networks.homoscedastic_multitask_loss module ReferentialGym.networks.networks module ReferentialGym.networks.residual_networks module ReferentialGym.referential_game module ReferentialGym.utils module ReferentialGym.utils.statsLogger module ReferentialGym.utils.statsLogger.statsLogger module ReferentialGym.utils.streamHandler module ReferentialGym.utils.streamHandler.StreamHandler module ReferentialGym.utils.utils module register() (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method) register_hook() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) reparameterize() (ReferentialGym.networks.autoregressive_networks.BetaVAE method) Rescale (class in ReferentialGym.datasets.CLEVR_dataset) (class in ReferentialGym.datasets.utils) RescaleNormalize (class in ReferentialGym.datasets.utils) reset() (ReferentialGym.agents.attention_lstm_cnn_listener.AttentionLSTMCNNListener method) (ReferentialGym.agents.caption_speaker.CaptionSpeaker method) (ReferentialGym.agents.categorical_obverter_agent.CategoricalObverterAgent method) (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent method) (ReferentialGym.agents.differentiable_relational_obverter.DifferentiableRelationalObverterAgent method) (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) (ReferentialGym.agents.eos_priored_lstm_cnn_speaker.EoSPrioredLSTMCNNSpeaker method) (ReferentialGym.agents.listener.Listener method) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener method) (ReferentialGym.agents.lstm_cnn_speaker.LSTMCNNSpeaker method) (ReferentialGym.agents.multi_head_lstm_cnn_speaker.MultiHeadLSTMCNNSpeaker method) (ReferentialGym.agents.obverter_agent.ObverterAgent method) (ReferentialGym.agents.speaker.Speaker method) (ReferentialGym.agents.transcoding_lstm_cnn_listener.TranscodingLSTMCNNListener method) (ReferentialGym.agents.transcoding_lstm_cnn_speaker.TranscodingLSTMCNNSpeaker method) (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method) ResizeNormalize (class in ReferentialGym.datasets.utils) ResNet (class in ReferentialGym.networks.residual_networks) ResNet18AvgPooledMHDPA (class in ReferentialGym.networks.residual_networks) ResNet18MHDPA (class in ReferentialGym.networks.residual_networks) ResNetAvgPooledEncoder (class in ReferentialGym.networks.autoregressive_networks) ResNetEncoder (class in ReferentialGym.networks.autoregressive_networks) ResNetParallelAttentionEncoder (class in ReferentialGym.networks.autoregressive_networks) ResNetPHDPAEncoder (class in ReferentialGym.networks.autoregressive_networks) retrieve_output_shape() (in module ReferentialGym.networks.networks)

S

sample() (ReferentialGym.agents.discriminative_listener.ExperienceBuffer method) (ReferentialGym.datasets.dataset.Dataset method) (ReferentialGym.datasets.dual_labeled_dataset.DualLabeledDataset method) (ReferentialGym.datasets.labeled_dataset.LabeledDataset method) (ReferentialGym.networks.autoregressive_networks.Bernoulli method) (ReferentialGym.networks.autoregressive_networks.Distribution method) (ReferentialGym.networks.autoregressive_networks.Normal method) save() (ReferentialGym.agents.agent.Agent method) (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) (ReferentialGym.networks.networks.MHDPA method) sentence_length_logging_hook() (in module ReferentialGym.agents.differentiable_obverter_agent) (in module ReferentialGym.agents.speaker) serve() (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method) set_mode() (ReferentialGym.datasets.dual_labeled_dataset.DualLabeledDataset method) (ReferentialGym.datasets.labeled_dataset.LabeledDataset method)

set_ref() (ReferentialGym.modules.current_agent_module.CurrentAgentModule method) setNbrDistractors() (ReferentialGym.datasets.dataset.Dataset method) shuffle() (in module ReferentialGym.datasets.dataset) SortOfCLEVRDataset (class in ReferentialGym.datasets.sort_of_CLEVR_dataset) Speaker (class in ReferentialGym.agents.speaker) SqueezeModule (class in ReferentialGym.modules.squeeze_module) statsLogger (class in ReferentialGym.utils.statsLogger.statsLogger) step() (ReferentialGym.networks.autoregressive_networks.TotalCorrelationDiscriminator method) StraightThroughGumbelSoftmaxLayer (class in ReferentialGym.utils.utils) StreamHandler (class in ReferentialGym.utils.streamHandler.StreamHandler) switch_epoch() (ReferentialGym.utils.statsLogger.statsLogger.statsLogger method) symbol_processing (ReferentialGym.agents.differentiable_obverter_agent.DifferentiableObverterAgent attribute) (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener attribute)

T

tau_fc (ReferentialGym.agents.lstm_cnn_listener.LSTMCNNListener attribute) ToLongTensor() (in module ReferentialGym.datasets.CLEVR_dataset) TopographicSimilarityMetricModule (class in ReferentialGym.modules.topographic_similarity_metric_module) TotalCorrelationDiscriminator (class in ReferentialGym.networks.autoregressive_networks)

ToTensor (class in ReferentialGym.datasets.CLEVR_dataset) train() (ReferentialGym.referential_game.ReferentialGame method) TranscodingLSTMCNNListener (class in ReferentialGym.agents.transcoding_lstm_cnn_listener) TranscodingLSTMCNNSpeaker (class in ReferentialGym.agents.transcoding_lstm_cnn_speaker)

U

UNet (class in ReferentialGym.networks.autoregressive_networks) UNetBlock (class in ReferentialGym.networks.autoregressive_networks)

update() (ReferentialGym.networks.autoregressive_networks.TotalCorrelationDiscriminator method) (ReferentialGym.utils.streamHandler.StreamHandler.StreamHandler method)

V

vae_loss_hook() (in module ReferentialGym.agents.agent)

VGG (class in ReferentialGym.networks.networks) VisualModule (class in ReferentialGym.modules.visual_module)

X

XSortOfCLEVRDataset (class in ReferentialGym.datasets.extended_sort_of_CLEVR_dataset)