hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update order of arch,params print

Browse Source
This commit is contained in:
Nicklas Hansen
2024-02-27 13:18:45 -08:00
parent d3bff48d58
commit c218c0ff1b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tdmpc2/trainer/base.py
View File

@@ -8,8 +8,8 @@ class Trainer:
self.buffer = buffer self.buffer = buffer
self.logger = logger self.logger = logger
if cfg.rank == 0: if cfg.rank == 0:
print("Learnable parameters: {:,}".format(self.agent.model.total_params))
print('Architecture:', self.agent.model) print('Architecture:', self.agent.model)
print("Learnable parameters: {:,}".format(self.agent.model.total_params))
def eval(self): def eval(self):
"""Evaluate a TD-MPC2 agent.""" """Evaluate a TD-MPC2 agent."""