File size: 406 Bytes
08cb485 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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
|