Pytorch_boilerplate