NWT-pytorch