How am I supposed to interpret it that anomaly mode is pointing to torch.nn.mse_loss() as the place where the gradient goes bad?
Similarly the graphing code posted by @tom above show red nodes from beginning (after input variable) to end (mse loss function).