Index

A

activation_cls (dataset2vec.config.Dataset2VecConfig attribute)

C

calculate_loss() (dataset2vec.model.Dataset2Vec method)

D

Dataset2Vec (class in dataset2vec.model) dataset2vec.config module dataset2vec.data module

dataset2vec.model module Dataset2VecConfig (class in dataset2vec.config) Dataset2VecLoader (class in dataset2vec.data)

F

f_block_repetitions (dataset2vec.config.Dataset2VecConfig attribute) f_dense_hidden_size (dataset2vec.config.Dataset2VecConfig attribute) f_out_size (dataset2vec.config.Dataset2VecConfig attribute)

f_res_hidden_size (dataset2vec.config.Dataset2VecConfig attribute) f_res_n_layers (dataset2vec.config.Dataset2VecConfig attribute) forward() (dataset2vec.model.Dataset2Vec method)

G

g_layers_sizes (dataset2vec.config.Dataset2VecConfig attribute)

gamma (dataset2vec.config.OptimizerConfig attribute)

H

h_block_repetitions (dataset2vec.config.Dataset2VecConfig attribute) h_dense_hidden_size (dataset2vec.config.Dataset2VecConfig attribute)

h_res_hidden_size (dataset2vec.config.Dataset2VecConfig attribute) h_res_n_layers (dataset2vec.config.Dataset2VecConfig attribute)

L

learning_rate (dataset2vec.config.OptimizerConfig attribute)

M

module dataset2vec.config dataset2vec.data dataset2vec.model

O

optimizer_cls (dataset2vec.config.OptimizerConfig attribute)

OptimizerConfig (class in dataset2vec.config) output_size (dataset2vec.config.Dataset2VecConfig attribute)

R

RepeatableDataset2VecLoader (class in dataset2vec.data)

W

weight_decay (dataset2vec.config.OptimizerConfig attribute)