|
|
|
@@ -69,6 +69,11 @@ class Deconvolution(ModifiedReLU): |
|
|
|
Args: |
|
|
|
network (Cell): The black-box model to be explained. |
|
|
|
|
|
|
|
Notes: |
|
|
|
The parsed `network` will be set to eval mode through `network.set_grad(False)` and `network.set_train(False)`. |
|
|
|
If you want to train the `network` afterwards, please reset it back to training mode through the opposite |
|
|
|
operations. |
|
|
|
|
|
|
|
Examples: |
|
|
|
>>> net = resnet50(10) |
|
|
|
>>> param_dict = load_checkpoint("resnet50.ckpt") |
|
|
|
@@ -98,6 +103,11 @@ class GuidedBackprop(ModifiedReLU): |
|
|
|
Args: |
|
|
|
network (Cell): The black-box model to be explained. |
|
|
|
|
|
|
|
Notes: |
|
|
|
The parsed `network` will be set to eval mode through `network.set_grad(False)` and `network.set_train(False)`. |
|
|
|
If you want to train the `network` afterwards, please reset it back to training mode through the opposite |
|
|
|
operations. |
|
|
|
|
|
|
|
Examples: |
|
|
|
>>> net = resnet50(10) |
|
|
|
>>> param_dict = load_checkpoint("resnet50.ckpt") |
|
|
|
|