Optim-wip: Update CustomModules tutorial for new changes#960

ProGamerGov wants to merge 2 commits intopytorch:optim-wippytorch/captum:optim-wipfrom

ProGamerGov:optim-wip-custom-modulesProGamerGov/captum:optim-wip-custom-modulesCopy head branch name to clipboard