fix(train): handle NHWC images in wandb logging#909

wadeKeith wants to merge 1 commit intoPhysical-Intelligence:mainPhysical-Intelligence/openpi:mainfrom

wadeKeith:fix/wandb-image-nhwc-formatwadeKeith/openpi-1:fix/wandb-image-nhwc-formatCopy head branch name to clipboard