update order of arch,params print

2024-02-27 13:18:45 -08:00
parent d3bff48d58
commit c218c0ff1b
1 changed files with 1 additions and 1 deletions
--- 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."""