from typing import Dict


def loss_dict() -> Dict[str, float]:
    """ Initialize loss dictionary for network sub-tasks
    """

    # Add z in front of each key name so they appear on the bottom (and together) in tensorboard
    d = {'z_normals_loss': 0.0, 'z_scene_loss': 0.0, 'z_depth_loss': 0.0, 'z_conf_loss': 0.0, 'z_loc_loss': 0.0}
    return d