Add custom objects loading parameter

raghakot / keras-vis

Neural network visualization toolkit for keras

https://raghakot.github.io/keras-vis

MIT License

2.98k stars 664 forks source link

Closed taneta closed 6 years ago

taneta commented 6 years ago

Add custom_objects parameter to apply_modifications function, which allows loading architectures with custom layers.

saksham-s commented 6 years ago

@raghakot @taneta this is working fine but the visualise_saliency method is giving an error due tot he custom layer.