Jensun Ravichandran 197b728c63

feat: add visualize method to visualization callbacks ...

All visualization callbacks now contain a `visualize` method that takes an
appropriate PyTorchLightning Module and visualizes it without the need for a
Trainer. This is to encourage users to perform one-off visualizations after
training.

2022-02-02 21:45:44 +01:00

..

models

feat: add visualize method to visualization callbacks

2022-02-02 21:45:44 +01:00