Update per_sample_grads.py

Author: svekars

intermediate_source/per_sample_grads.py

@@ -168,7 +168,6 @@ def compute_loss(params, buffers, sample, target):
 # we can double check that the results using ``grad`` and ``vmap`` match the
 # results of hand processing each one individually:
 
-# Replace the comparison section with this updated code
 for name, ft_per_sample_grad in ft_per_sample_grads.items():
     # Find the corresponding manually computed gradient
     idx = list(model.named_parameters()).index((name, model.get_parameter(name)))