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,
|
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
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user