mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 19:20:19 +01:00
Add is_on_policy property to agents (#480)
This commit is contained in:
@@ -31,6 +31,10 @@ class ImitationAgent(Agent):
|
||||
super().__init__(agent_parameters, parent)
|
||||
self.imitation = True
|
||||
|
||||
@property
|
||||
def is_on_policy(self) -> bool:
|
||||
return False
|
||||
|
||||
def extract_action_values(self, prediction):
|
||||
return prediction.squeeze()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user