From c218c0ff1b39ef03533636f0b5ca869477a108b9 Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Tue, 27 Feb 2024 13:18:45 -0800 Subject: [PATCH] update order of arch,params print --- tdmpc2/trainer/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tdmpc2/trainer/base.py b/tdmpc2/trainer/base.py index 3c4d0b1..88f0167 100755 --- a/tdmpc2/trainer/base.py +++ b/tdmpc2/trainer/base.py @@ -8,8 +8,8 @@ class Trainer: self.buffer = buffer self.logger = logger if cfg.rank == 0: - print("Learnable parameters: {:,}".format(self.agent.model.total_params)) print('Architecture:', self.agent.model) + print("Learnable parameters: {:,}".format(self.agent.model.total_params)) def eval(self): """Evaluate a TD-MPC2 agent."""