Refactor modules (#7823)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-05-15 19:44:16 +02:00
1 parent f00071416f
commit 8aa2085a7e
3 files changed
+17 -23

No files matched your search

+1 -1
View File
@@ -197,7 +197,7 @@ class Model(nn.Module):
m(x.copy() if c else x)
dt.append((time_sync() - t) * 100)
if m == self.model[0]:
LOGGER.info(f"{'time (ms)':>10s} {'GFLOPs':>10s} {'params':>10s} {'module'}")
LOGGER.info(f"{'time (ms)':>10s} {'GFLOPs':>10s} {'params':>10s} module")
LOGGER.info(f'{dt[-1]:10.2f} {o:10.2f} {m.np:10.0f} {m.type}')
if c:
LOGGER.info(f"{sum(dt):10.2f} {'-':>10s} {'-':>10s} Total")