jakegrigsby commited on
Commit
143bdb3
·
verified ·
1 Parent(s): d2d5422

Delete config.txt

Browse files
Files changed (1) hide show
  1. config.txt +0 -121
config.txt DELETED
@@ -1,121 +0,0 @@
1
- # Parameters for Actor:
2
- # ==============================================================================
3
- Actor.activation = 'leaky_relu'
4
- Actor.cont_dist_kind = 'normal'
5
- Actor.d_hidden = 512
6
- Actor.dropout_p = 0.0
7
- Actor.gmm_modes = 5
8
- Actor.log_std_high = 2.0
9
- Actor.log_std_low = -5.0
10
- Actor.n_layers = 2
11
-
12
- # Parameters for Agent:
13
- # ==============================================================================
14
- Agent.fake_filter = False
15
- Agent.gamma = 0.999
16
- Agent.num_critics = 6
17
- Agent.num_critics_td = 2
18
- Agent.popart = True
19
- Agent.reward_multiplier = 10.0
20
- Agent.tau = 0.003
21
- Agent.use_multigamma = True
22
- Agent.use_target_actor = True
23
-
24
- # Parameters for Experiment:
25
- # ==============================================================================
26
- Experiment.batches_per_update = 1
27
- Experiment.critic_loss_weight = 10.0
28
- Experiment.env_mode = 'async'
29
- Experiment.force_reset_train_envs_every = None
30
- Experiment.grad_clip = 1.0
31
- Experiment.has_replay_buffer_rights = True
32
- Experiment.l2_coeff = 0.001
33
- Experiment.learning_rate = 0.0001
34
- Experiment.local_time_optimizer = False
35
- Experiment.lr_warmup_steps = 500
36
- Experiment.mixed_precision = 'no'
37
- Experiment.padded_sampling = 'none'
38
- Experiment.save_trajs_as = 'npz'
39
- Experiment.stagger_traj_file_lengths = True
40
- Experiment.wandb_group_name = None
41
-
42
- # Parameters for FlashAttention:
43
- # ==============================================================================
44
- FlashAttention.window_size = (-1, -1)
45
-
46
- # Parameters for MetamonTstepEncoder:
47
- # ==============================================================================
48
- MetamonTstepEncoder.d_model = 160
49
- MetamonTstepEncoder.extra_emb_dim = 18
50
- MetamonTstepEncoder.n_heads = 8
51
- MetamonTstepEncoder.n_layers = 5
52
- MetamonTstepEncoder.scratch_tokens = 11
53
- MetamonTstepEncoder.token_mask_aug = False
54
-
55
- # Parameters for Multigammas:
56
- # ==============================================================================
57
- Multigammas.continuous = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
58
- Multigammas.discrete = [0.1, 0.9, 0.95, 0.97, 0.99, 0.995]
59
-
60
- # Parameters for MultiModalEmbedding:
61
- # ==============================================================================
62
- MultiModalEmbedding.dropout = 0.05
63
- MultiModalEmbedding.numerical_tokens = 6
64
-
65
- # Parameters for MultiTaskAgent:
66
- # ==============================================================================
67
- MultiTaskAgent.fbc_filter_k = 3
68
- MultiTaskAgent.offline_coeff = 1.0
69
- MultiTaskAgent.online_coeff = 0.0
70
-
71
- # Parameters for NCritics:
72
- # ==============================================================================
73
- NCritics.activation = 'leaky_relu'
74
- NCritics.d_hidden = 512
75
- NCritics.dropout_p = 0.0
76
- NCritics.n_layers = 2
77
-
78
- # Parameters for NCriticsTwoHot:
79
- # ==============================================================================
80
- NCriticsTwoHot.activation = 'leaky_relu'
81
- NCriticsTwoHot.d_hidden = 512
82
- NCriticsTwoHot.dropout_p = 0.0
83
- NCriticsTwoHot.max_return = 1100
84
- NCriticsTwoHot.min_return = -1100
85
- NCriticsTwoHot.n_layers = 2
86
- NCriticsTwoHot.output_bins = 96
87
- NCriticsTwoHot.use_symlog = False
88
-
89
- # Parameters for PopArtLayer:
90
- # ==============================================================================
91
- PopArtLayer.beta = 0.0005
92
- PopArtLayer.init_nu = 100.0
93
-
94
- # Parameters for TformerTrajEncoder:
95
- # ==============================================================================
96
- TformerTrajEncoder.activation = 'leaky_relu'
97
- TformerTrajEncoder.causal = True
98
- TformerTrajEncoder.d_ff = 5120
99
- TformerTrajEncoder.d_model = 1280
100
- TformerTrajEncoder.dropout_attn = 0.0
101
- TformerTrajEncoder.dropout_emb = 0.05
102
- TformerTrajEncoder.dropout_ff = 0.05
103
- TformerTrajEncoder.dropout_qkv = 0.0
104
- TformerTrajEncoder.head_scaling = True
105
- TformerTrajEncoder.n_heads = 20
106
- TformerTrajEncoder.n_layers = 9
107
- TformerTrajEncoder.norm = 'layer'
108
- TformerTrajEncoder.normformer_norms = True
109
- TformerTrajEncoder.sigma_reparam = True
110
-
111
- # Parameters for TimestepTransformer:
112
- # ==============================================================================
113
- # None.
114
-
115
- # Parameters for TokenEmbedding:
116
- # ==============================================================================
117
- # None.
118
-
119
- # Parameters for TransformerTurnEmbedding:
120
- # ==============================================================================
121
- TransformerTurnEmbedding.dropout = 0.05