PyTorch Forums
marc_j