prototorch_models/prototorch
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