Optim-wip: Fix _check_layer_in_model test#606

NarineK merged 1 commit intometa-pytorch:optim-wipmeta-pytorch/captum:optim-wipfrom

ProGamerGov:patch-2ProGamerGov/captum:patch-2Copy head branch name to clipboard