Skip to content

Commit d365ff9

Browse files
committed
update bafc_dmc_conf to exp094
1 parent 688f053 commit d365ff9

File tree

1 file changed

+16
-14
lines changed

1 file changed

+16
-14
lines changed

alf/examples/bafc_dmc_conf.py

Lines changed: 16 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,20 @@
2222
from alf.examples.benchmarks.dm_control import dmc_conf
2323
from alf.optimizers import Adam
2424

25-
actor_hidden_layers = (256, 256)
26-
joint_hidden_layers = (256, 256)
27-
# actor_hidden_layers = (32, 32)
28-
# joint_hidden_layers = (32, 32)
25+
debug_mode = False
2926
optimizer = Adam(lr=5e-4)
3027
use_obs_normalizer = True
3128
obs_normalizer_clipping = False
3229

30+
if debug_mode:
31+
actor_hidden_layers = (32, 32)
32+
joint_hidden_layers = (32, 32)
33+
num_actor_eval_samples = 64
34+
else:
35+
actor_hidden_layers = (256, 256)
36+
joint_hidden_layers = (256, 256)
37+
num_actor_eval_samples = 512
38+
3339
if use_obs_normalizer:
3440
data_transformer_ctor = ObservationNormalizer
3541
else:
@@ -56,24 +62,20 @@
5662
actor_network_cls=actor_network_cls,
5763
critic_network_cls=critic_network_cls,
5864
num_actors=10,
59-
use_target_actor=False,
6065
use_bootstrap_actors=True,
6166
bootstrap_mask_prob=0.8,
62-
# num_actor_eval_samples=512,
63-
num_actor_eval_samples=64,
67+
num_actor_eval_samples=num_actor_eval_samples,
6468
eval_samples_init_method='normal',
6569
eval_samples_clipping=obs_normalizer_clipping,
66-
actor_eval_type='last',
67-
actor_encoding_dim=256,
70+
actor_eval_type='exclude_input',
71+
actor_encoding_dim=None,
6872
obs_action_encoding_dim=128,
6973
actor_utd=1,
70-
critic_utd=5,
74+
critic_utd=2,
75+
critic_respect_exp_batch_size=True,
7176
target_critic_tau=0.005,
7277
target_critic_period=1,
73-
target_critic_use_ema=False,
74-
target_actor_tau=0.05,
75-
target_actor_period=1,
76-
target_actor_use_ema=False)
78+
target_critic_use_ema=False)
7779

7880
alf.config(
7981
'TransformerEncoder',

0 commit comments

Comments
 (0)