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 | |
cvx_up: true | |
decoder_dropout: 0.0 | |
depth: 8 | |
feat_dim: 220 | |
feat_ver: 1 | |
final_dim: 64 | |
im_channels: 64 | |
img_aug_2: 0.05 | |
img_feat_dim: 3 | |
img_patch_size: 14 | |
img_size: 224 | |
inp_pre_con: false | |
inp_pre_pro: false | |
lang_dim: 512 | |
lang_len: 77 | |
norm_corr: true | |
num_rot: 72 | |
pe_fix: true | |
proprio_dim: 4 | |
rend_three_views: true | |
rot_ver: 1 | |
self_cross_ver: 1 | |
st_sca: 4 | |
st_wpt_loc_aug: 0.05 | |
st_wpt_loc_inp_no_noise: true | |
stage_two: true | |
use_point_renderer: true | |
weight_tie_layers: false | |
wpt_img_aug: 0.0 | |
xops: true | |