Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Improve extreme batch visualization callbacks (#1488)
* Improve visualization callbacks * Fix docstrings * Added tests and improved documentation to reflect what is the expected return type and layout (channels last) of the image tensor for visualization
- Loading branch information