Cannot currently use the Wandb image logging that worked in LitJTML_Seg because the current images have 128(=num_keypoints) channels. I've commented out the callbacks from validation and test batch end and an imaging step in the validation_step method in kp_pose_hrnet_module.py .
Cannot currently use the Wandb image logging that worked in LitJTML_Seg because the current images have 128(=num_keypoints) channels. I've commented out the callbacks from validation and test batch end and an imaging step in the validation_step method in kp_pose_hrnet_module.py .