(logger, mode, scalar_dict, global_step)
| 73 | |
| 74 | |
| 75 | def save_scalars(logger, mode, scalar_dict, global_step): |
| 76 | scalar_dict = tensor2float(scalar_dict) |
| 77 | for key, value in scalar_dict.items(): |
| 78 | if not isinstance(value, (list, tuple)): |
| 79 | name = '{}/{}'.format(mode, key) |
| 80 | logger.add_scalar(name, value, global_step) |
| 81 | else: |
| 82 | for idx in range(len(value)): |
| 83 | name = '{}/{}_{}'.format(mode, key, idx) |
| 84 | logger.add_scalar(name, value[idx], global_step) |
| 85 | |
| 86 | |
| 87 | def save_images(logger, mode, images_dict, global_step): |
no test coverage detected