Update bnaf.py - Githubissues

nicola-decao / BNAF

Pytorch implementation of Block Neural Autoregressive Flow

http://arxiv.org/abs/1904.04676

MIT License

176 stars 34 forks source link

Update bnaf.py #4

Closed gauenk closed 4 years ago

gauenk commented 4 years ago

this update will fix the warning: "[W IndexingUtils.h:20] Warning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead. (function expandTensors)"

nicola-decao commented 4 years ago

Seems good to me! Thanks 😊