MCPcopy Create free account
hub / github.com/MotrixLab/MotionDiffuse / print_current_loss

Function print_current_loss

text2motion/utils/utils.py:36–58  ·  view source on GitHub ↗
(start_time, niter_state, losses, epoch=None, inner_iter=None)

Source from the content-addressed store, hash-verified

34
35
36def print_current_loss(start_time, niter_state, losses, epoch=None, inner_iter=None):
37
38 def as_minutes(s):
39 m = math.floor(s / 60)
40 s -= m * 60
41 return '%dm %ds' % (m, s)
42
43 def time_since(since, percent):
44 now = time.time()
45 s = now - since
46 es = s / percent
47 rs = es - s
48 return '%s (- %s)' % (as_minutes(s), as_minutes(rs))
49
50 if epoch is not None:
51 print('epoch: %3d niter: %6d inner_iter: %4d' % (epoch, niter_state, inner_iter), end=" ")
52
53 now = time.time()
54 message = '%s'%(as_minutes(now - start_time))
55
56 for k, v in losses.items():
57 message += ' %s: %.4f ' % (k, v)
58 print(message)
59
60
61def compose_gif_img_list(img_list, fp_out, duration):

Callers 1

trainMethod · 0.90

Calls 1

as_minutesFunction · 0.85

Tested by

no test coverage detected