Typo in do_train

[deepkyu]

Hi,
it's so late but congrats on great work!

In the below,

litepose/lib/core/trainer.py

Lines 139 to 154 in c3e4ba0

	for idx in range(cfg.LOSS.NUM_STAGES):
	writer.add_scalar(
	'train_stage{}_heatmaps_loss'.format(i),
	heatmaps_loss_meter[idx].val,
	global_steps
	)
	writer.add_scalar(
	'train_stage{}_push_loss'.format(idx),
	push_loss_meter[idx].val,
	global_steps
	)
	writer.add_scalar(
	'train_stage{}_pull_loss'.format(idx),
	pull_loss_meter[idx].val,
	global_steps
	)

i at 'train_stage{}_heatmaps_loss'.format(i), seems to be changed to idx.

Or, just in case, is there any specific reason for logging in each frequency?

Really appreciate any help you can provide :)