PyTorch Forums
maris_ancans