Skip to content

Commit f92edc7

Browse files
authored
Updating Recorder to accomodate "solved" key (#833)
1 parent 9003a56 commit f92edc7

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

torchrl/trainers/trainers.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1157,8 +1157,6 @@ def __call__(self, batch: TensorDictBase) -> Dict:
11571157
out[self.out_keys[key]] = mean_value
11581158
out["total_" + self.out_keys[key]] = total_value
11591159
continue
1160-
if key == "solved":
1161-
value = value.any().float()
11621160
out[self.out_keys[key]] = value
11631161
out["log_pbar"] = self.log_pbar
11641162
self._count += 1

0 commit comments

Comments
 (0)