CUDA now Supported

CUDA now supported

v1.2.4

Bug Fix + Added option to get Label from Layer

v1.2.2

Release Notes - PyTorchLayerViz v1.2.2

Added Features

• You have now the option to not print the Images when using the function get_feature_maps()
• Unittests have been added

Release Notes - PyTorchLayerViz v1.2.1

Release Notes - PyTorchLayerViz v1.2.1

Bug Fixes

• The Function don't crashes anymore when giving a colorful input

v1.2

Release Notes - PyTorchLayerViz v1.2

What's New

• Return Pictures: The get_feature_maps() function now returns all feature maps as NumPy arrays.
• Colorful Picture: The get_feature_maps() function now accepts color images (3-channel RGB).

1.1

Release Notes - PyTorchLayerViz v1.1

What's New

• README File: Added a comprehensive README file to help users understand the purpose of the project, how to install it, and how to use it effectively.

Bug Fixes

• Import Bug: I added the imports to the main.py file