fixed warning (#464)
This commit is contained in:
committed by
pzhokhov
parent
43ed76944b
commit
1e9051e87e
@@ -116,7 +116,7 @@ class RolloutWorker:
|
|||||||
return self.generate_rollouts()
|
return self.generate_rollouts()
|
||||||
|
|
||||||
if np.isnan(o_new).any():
|
if np.isnan(o_new).any():
|
||||||
self.logger.warning('NaN caught during rollout generation. Trying again...')
|
self.logger.warn('NaN caught during rollout generation. Trying again...')
|
||||||
self.reset_all_rollouts()
|
self.reset_all_rollouts()
|
||||||
return self.generate_rollouts()
|
return self.generate_rollouts()
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user