ICLR OSC 2022
Contents:
Introduction
Setup
API Reference
osc.data
osc.loss_global
osc.loss_objects
osc.lr_scheduler
osc.models
osc.train
osc.utils
osc.viz
osc.viz.attentions
osc.viz.attentions.viz_slot_attns
osc.viz.attentions.viz_slot_rollout_all_options
osc.viz.attentions.viz_vit_attns
osc.viz.attentions.viz_vit_rollout_all_options
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
ICLR OSC 2022
»
osc
»
osc.viz
»
osc.viz.attentions
»
osc.viz.attentions.viz_vit_attns
View page source
osc.viz.attentions.viz_vit_attns
viz_vit_attns
(
images
,
attns_dict
,
head_reduction
=
'mean'
,
adjust_residual
=
True
)
[source]