PyTorch Forums
cdancette