https://github.com/carolineec/informative-drawings/blob/2788888c37df5f6c447849f8300b6d5cdc2dab79/train.py#L336 https://github.com/carolineec/informative-drawings/blob/2788888c37df5f6c447849f8300b6d5cdc2dab79/train.py#L337 In lines 336 and 337 referenced above, I think it should be myloss_recon instead of loss_recog. Please let me know if I am missing something
