From fb07cdac3f0934a6c3977b79e25f0db312f7b695 Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Sun, 10 Nov 2024 12:27:24 -0800 Subject: [PATCH] update compile print --- tdmpc2/tdmpc2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tdmpc2/tdmpc2.py b/tdmpc2/tdmpc2.py index 8e6b435..e4d8ec2 100755 --- a/tdmpc2/tdmpc2.py +++ b/tdmpc2/tdmpc2.py @@ -36,7 +36,7 @@ class TDMPC2(torch.nn.Module): ) if self.cfg.multitask else self._get_discount(cfg.episode_length) self._prev_mean = torch.nn.Buffer(torch.zeros(self.cfg.horizon, self.cfg.action_dim, device=self.device)) if cfg.compile: - print('compiling - update') + print('Compiling update function with torch.compile...') self._update = torch.compile(self._update, mode="reduce-overhead") @property