Uncertainty loss balancing logvars not included in checkpoint ?
It looks like the logvars of Uncertainty loss balancing are not included in the checkpoint, or are not loaded when a checkpoint is loaded to continue training:
Here, checkpoint corresponding to step 29 is loaded