iris-hero / config.json
ruffy369's picture
Update config.json
ef35fd0 verified
{
"_name_or_path": "ruffy369/iris-hero",
"architectures": [
"IrisModel"
],
"attention": "causal",
"attn_pdrop": 0.1,
"attn_resolutions": [
8,
16
],
"bos_token_id": 50256,
"ch": 64,
"ch_mult": [
1,
1,
1,
1,
1
],
"device": "cuda:0",
"dropout": 0.0,
"embed_dim_discrete_autoencoder": 512,
"embed_dim_world_model": 256,
"embed_pdrop": 0.1,
"entropy_weight": 0.001,
"eos_token_id": 50256,
"gamma": 0.995,
"grad_acc_steps_actor_critic": 1,
"grad_acc_steps_discrete_autoencoder": 1,
"grad_acc_steps_world_model": 1,
"imagine_horizon_train_actor_critic": 20,
"in_channels": 3,
"initializer_range": 0.02,
"lambda_": 0.95,
"max_blocks": 20,
"max_tokens": 340,
"model_type": "iris",
"num_actions": 18,
"num_heads": 4,
"num_layers": 10,
"num_res_blocks": 2,
"out_ch": 3,
"pad_token_id": 1,
"resid_pdrop": 0.1,
"resolution": 64,
"sequence_length": 20,
"tokens_per_block": 17,
"transformers_version": "4.44.0.dev0",
"use_cache": true,
"use_original_obs_actor_critic": false,
"vocab_size": 512,
"z_channels": 512
}