ICLR OSC 2022
Contents:
Introduction
Setup
API Reference
ICLR OSC 2022
»
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
_
__add__() (ModelOutput method)
__call__() (AugmentFnStateless method)
(CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(ModelLoss method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
__mul__() (ModelOutput method)
A
add_module() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
apply() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
array_to_pil() (in module osc.viz.utils)
augment_center_crop() (in module osc.data.utils)
augment_train() (in module osc.data.utils)
augment_twice() (in module osc.data.utils)
AugmentFnStateless (class in osc.data.utils)
AverageMetric (class in osc.utils)
B
batched_otsu() (in module osc.viz.utils)
batched_otsu_pt() (in module osc.viz.utils)
batches_per_epoch() (in module osc.utils)
batches_per_epoch_test() (in module osc.train)
batches_per_epoch_train() (in module osc.train)
batches_per_epoch_val() (in module osc.train)
bfloat16() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
buffers() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
build_dataset_test() (in module osc.train)
build_dataset_train() (in module osc.train)
build_dataset_val() (in module osc.train)
build_dataset_vqa() (in module osc.train)
build_linear_probe() (in module osc.train)
build_loss_fn_global() (in module osc.train)
build_loss_fn_objects() (in module osc.train)
build_model() (in module osc.train)
build_optimizer() (in module osc.train)
build_scheduler() (in module osc.train)
C
children() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
CoAttentionBlock (class in osc.models.attentions)
compute() (AverageMetric method)
contrastive_loss_best_worst() (in module osc.loss_global)
count() (ModelOutput method)
cpu() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
cross_attn_rollout() (in module osc.viz.rollout)
CrossAttention (class in osc.models.attentions)
CrossAttentionBlock (class in osc.models.attentions)
CrossAttentionDecoder (class in osc.models.attentions)
cuda() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
D
deserialize_image() (in module osc.data.tfrecords)
deserialize_image_and_mask() (in module osc.data.tfrecords)
double() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
E
eval() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
exponential_lr_gamma() (in module osc.lr_scheduler)
extra_repr() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
extract_vqa_features() (in module osc.train)
F
f_backbone (ModelOutput attribute)
f_global (ModelOutput attribute)
f_slots (ModelOutput attribute)
fig_save_display() (in module osc.viz.utils)
fill_diagonal_() (in module osc.utils)
filter_cfg_for_wandb() (in module osc.wandb_utils)
find_run_by_name() (in module osc.wandb_utils)
fix_tf_dtypes() (in module osc.data.clevr_with_masks)
(in module osc.data.multi_dsprites)
(in module osc.data.objects_room)
(in module osc.data.tetrominoes)
float() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
forward() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
forward_with_attns() (in module osc.models.models)
G
get_buffer() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
get_extra_state() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
get_iterator() (in module osc.data.clevr_with_masks)
(in module osc.data.multi_dsprites)
(in module osc.data.objects_room)
(in module osc.data.tetrominoes)
get_parameter() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
get_params() (in module osc.data.random_resized_crop)
get_submodule() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
get_viz_batch() (in module osc.train)
H
half() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
history_groupby_name() (in module osc.viz.history)
I
img_html() (in module osc.viz.utils)
img_hwc_to_chw() (in module osc.data.utils)
index() (ModelOutput method)
init_weights() (SampledObjectTokens method)
K
kmeans_backbone() (in module osc.viz.backbone)
kmeans_clusters() (in module osc.viz.backbone)
KmeansCosineObjectTokens (class in osc.models.embeds)
KmeansEuclideanObjectTokens (class in osc.models.embeds)
L
latest_checkpoint() (in module osc.utils)
LearnedObjectTokens (class in osc.models.embeds)
LinearWarmupCosineAnneal (class in osc.lr_scheduler)
load_state_dict() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
log_env_info() (in module osc.train)
log_model_parameters() (in module osc.train)
M
main() (in module osc.data.clevr_with_masks)
(in module osc.train)
make_grid_pil() (in module osc.viz.utils)
match_objects() (in module osc.viz.loss_objects)
match_segmentation_masks() (in module osc.viz.segmentation)
matching_contrastive_loss() (in module osc.loss_objects)
matching_contrastive_loss_best_worst() (in module osc.loss_objects)
matching_contrastive_loss_per_img() (in module osc.loss_objects)
matching_similarity_loss_per_img() (in module osc.loss_objects)
MLP (class in osc.models.utils)
Model (class in osc.models.models)
ModelLoss (class in osc.train)
ModelOutput (class in osc.models.models)
module
osc
osc.data
osc.data.clevr_with_masks
osc.data.iterable_dataloader
osc.data.multi_dsprites
osc.data.objects_room
osc.data.random_resized_crop
osc.data.tetrominoes
osc.data.tfrecords
osc.data.utils
osc.loss_global
osc.loss_objects
osc.lr_scheduler
osc.models
osc.models.attentions
osc.models.embeds
osc.models.models
osc.models.utils
osc.models.vit
osc.train
osc.utils
osc.viz
osc.viz.attentions
osc.viz.backbone
osc.viz.embeds
osc.viz.history
osc.viz.loss_global
osc.viz.loss_objects
osc.viz.rollout
osc.viz.segmentation
osc.viz.utils
osc.wandb_utils
modules() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
MyIterableDataset (class in osc.data.iterable_dataloader)
N
named_buffers() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
named_children() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
named_modules() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
named_parameters() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
normalize_tf() (in module osc.data.utils)
O
osc
module
osc.data
module
osc.data.clevr_with_masks
module
osc.data.iterable_dataloader
module
osc.data.multi_dsprites
module
osc.data.objects_room
module
osc.data.random_resized_crop
module
osc.data.tetrominoes
module
osc.data.tfrecords
module
osc.data.utils
module
osc.loss_global
module
osc.loss_objects
module
osc.lr_scheduler
module
osc.models
module
osc.models.attentions
module
osc.models.embeds
module
osc.models.models
module
osc.models.utils
module
osc.models.vit
module
osc.train
module
osc.utils
module
osc.viz
module
osc.viz.attentions
module
osc.viz.backbone
module
osc.viz.embeds
module
osc.viz.history
module
osc.viz.loss_global
module
osc.viz.loss_objects
module
osc.viz.rollout
module
osc.viz.segmentation
module
osc.viz.utils
module
osc.wandb_utils
module
P
p_global (ModelOutput attribute)
p_slots (ModelOutput attribute)
parameters() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
PositionalEmbedding (class in osc.models.embeds)
prepare_test_segmentation() (in module osc.data.clevr_with_masks)
prepare_test_vqa() (in module osc.data.clevr_with_masks)
print_arrays() (in module osc.utils)
R
random_resized_crop() (in module osc.data.random_resized_crop)
ravel_multi_index_tf() (in module osc.utils)
register_backward_hook() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
register_buffer() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
register_forward_hook() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
register_forward_pre_hook() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
register_full_backward_hook() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
register_parameter() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
reinforce_type() (MyIterableDataset method)
remove_xyticks() (in module osc.viz.utils)
requires_grad_() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
run_test_linear_probes() (in module osc.train)
run_test_segmentation() (in module osc.train)
run_train_epoch() (in module osc.train)
run_train_val_viz_epochs() (in module osc.train)
run_val_epoch() (in module osc.train)
run_viz() (in module osc.train)
S
SampledObjectTokens (class in osc.models.embeds)
seed_everything() (in module osc.utils)
self_attn_rollout() (in module osc.viz.rollout)
SelfAttention (class in osc.models.attentions)
SelfAttentionBlock (class in osc.models.attentions)
serialize_image() (in module osc.data.tfrecords)
serialize_image_and_mask() (in module osc.data.tfrecords)
set_extra_state() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
setup_wandb() (in module osc.wandb_utils)
share_memory() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
show_sample() (in module osc.data.clevr_with_masks)
(in module osc.data.multi_dsprites)
(in module osc.data.objects_room)
(in module osc.data.tetrominoes)
SigIntCatcher (class in osc.utils)
slot_attn_rollout() (in module osc.viz.rollout)
SlotAttention (class in osc.models.attentions)
state_dict() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
step() (StepCounter method)
StepCounter (class in osc.utils)
subplots_grid() (in module osc.viz.utils)
T
text_html() (in module osc.viz.utils)
to() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
to_dotlist() (in module osc.utils)
to_empty() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
torch_kmeans_cosine() (in module osc.models.embeds)
torch_kmeans_euclidean() (in module osc.models.embeds)
train() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
type() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
U
unnormalize_pt() (in module osc.data.utils)
update() (AverageMetric method)
update_cfg() (in module osc.train)
V
ViTBackbone (class in osc.models.vit)
viz_contrastive_loss_global() (in module osc.viz.loss_global)
viz_contrastive_loss_global_probs() (in module osc.viz.loss_global)
viz_contrastive_loss_objects() (in module osc.viz.loss_objects)
viz_contrastive_loss_objects_probs() (in module osc.viz.loss_objects)
viz_history() (in module osc.viz.history)
viz_positional_embedding() (in module osc.viz.embeds)
viz_slot_attns() (in module osc.viz.attentions)
viz_slot_rollout_all_options() (in module osc.viz.attentions)
viz_vit_attns() (in module osc.viz.attentions)
viz_vit_rollout_all_options() (in module osc.viz.attentions)
W
wrap_with_seed() (in module osc.data.utils)
X
xpu() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)
Z
zero_grad() (CoAttentionBlock method)
(CrossAttention method)
(CrossAttentionBlock method)
(CrossAttentionDecoder method)
(KmeansCosineObjectTokens method)
(KmeansEuclideanObjectTokens method)
(LearnedObjectTokens method)
(MLP method)
(Model method)
(PositionalEmbedding method)
(SampledObjectTokens method)
(SelfAttention method)
(SelfAttentionBlock method)
(SlotAttention method)
(ViTBackbone method)