CriticFactoryReuseActor: Fix the case where we want to reuse an actor's
preprocessing network for the critic (must be applied before concatenating the actions)
This commit is contained in:
parent
0b494845c9
commit
d18ded333e
@ -197,7 +197,11 @@ class CriticFactoryReuseActor(CriticFactory):
|
||||
last_size=last_size,
|
||||
).to(device)
|
||||
elif envs.get_type().is_continuous():
|
||||
return continuous.Critic(actor.get_preprocess_net(), device=device).to(device)
|
||||
return continuous.Critic(
|
||||
actor.get_preprocess_net(),
|
||||
device=device,
|
||||
apply_preprocess_net_to_obs_only=True,
|
||||
).to(device)
|
||||
else:
|
||||
raise ValueError
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user