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:
Dominik Jain 2024-04-29 14:09:48 +02:00
parent 0b494845c9
commit d18ded333e

View File

@ -197,7 +197,11 @@ class CriticFactoryReuseActor(CriticFactory):
last_size=last_size, last_size=last_size,
).to(device) ).to(device)
elif envs.get_type().is_continuous(): 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: else:
raise ValueError raise ValueError