title: How to use different data augmentation for Subsets in PyTorch
body: <p>How to use different data augmentation (transforms) for different <code>Subset</code>s in PyTorch?</p>
  
  <p>For instance:</p>
  
  <pre><code>train, test = torch.utils.data.random_split(dataset, [80000, 2000])
  </code></pre>
  
  <p><code>train</code> and <code>test</code> will have the same transforms as <code>dataset</code>. How to use custom transforms for these subsets?</p>