| activation: lrelu | |
| add_corr: true | |
| add_depth: true | |
| add_lang: true | |
| add_pixel_loc: true | |
| add_proprio: true | |
| attn_dim: 512 | |
| attn_dim_head: 64 | |
| attn_dropout: 0.1 | |
| attn_heads: 8 | |
| decoder_dropout: 0.0 | |
| depth: 8 | |
| feat_dim: 220 | |
| final_dim: 64 | |
| im_channels: 64 | |
| img_feat_dim: 3 | |
| img_patch_size: 11 | |
| img_size: 220 | |
| lang_dim: 512 | |
| lang_len: 77 | |
| pe_fix: true | |
| proprio_dim: 3 | |
| self_cross_ver: 1 | |
| use_dropout: true | |
| weight_tie_layers: false | |