(self, config, lr)
| 20 | |
| 21 | class TCP_planner(pl.LightningModule): |
| 22 | def __init__(self, config, lr): |
| 23 | super().__init__() |
| 24 | self.lr = lr |
| 25 | self.config = config |
| 26 | self.model = TCP(config) |
| 27 | self._load_weight() |
| 28 | |
| 29 | def _load_weight(self): |
| 30 | rl_state_dict = torch.load(self.config.rl_ckpt, map_location='cpu')['policy_state_dict'] |
nothing calls this directly
no test coverage detected