diff --git a/ding/model/template/vac.py b/ding/model/template/vac.py index 948ff3b9c39bd23934ea3a3b019a6d828c73a72a..d776259ee70ce93375326ed93a7ad9ca3c8f66e0 100644 --- a/ding/model/template/vac.py +++ b/ding/model/template/vac.py @@ -25,7 +25,7 @@ class VAC(nn.Module): continuous: bool = False, encoder_hidden_size_list: SequenceType = [128, 128, 64], actor_head_hidden_size: int = 64, - actor_head_layer_num: int = 2, + actor_head_layer_num: int = 1, critic_head_hidden_size: int = 64, critic_head_layer_num: int = 1, activation: Optional[nn.Module] = nn.ReLU(),