Skip to content

Commit

Permalink
Update batch size notation to be consistent
Browse files Browse the repository at this point in the history
Summary: fix notation of batch_size in RC safety module

Reviewed By: rodrigodesalvobraz

Differential Revision: D54178304

fbshipit-source-id: 9b4eefdb25dd3ecb81d895b53206840915a18385
  • Loading branch information
Yonathan Efroni authored and facebook-github-bot committed Feb 26, 2024
1 parent 611928a commit b3b6005
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pearl/safety_modules/reward_constrained_safety_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def __init__(
self.lr_lambda = lr_lambda
self.lambda_constraint_ub_value = lambda_constraint_ub_value
self.lambda_constraint = lambda_constraint_init_value
self.batch_size = batch_size
self._batch_size = batch_size

# initialize parameters of cost critic
assert hasattr(action_space, "action_dim")
Expand Down

0 comments on commit b3b6005

Please sign in to comment.