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

ProGamerGov wants to merge 2 commits intometa-pytorch:optim-wipmeta-pytorch/captum:optim-wipfrom

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