First, thanks a lot for providing this package :)
Currently, the requirements.txt, and thus the dependencies in the setup.py are for very specific versions of Pytorch etc. This shouldn't be the case if you want this package to be used as a general library (think of a second package that would do the same but ask for an incompatible version of PyTorch and would prevent any possible installation of the two together). The end user might also be needing a more recent version of PyTorch.
Given that PyTorch is almost always backward compatible, and quite stable, I think the requirements for it could be changed from ==1.8.1 to >=1.8.1. I believe the same would be true for the other packages.
First, thanks a lot for providing this package :)
Currently, the requirements.txt, and thus the dependencies in the
setup.pyare for very specific versions of Pytorch etc. This shouldn't be the case if you want this package to be used as a general library (think of a second package that would do the same but ask for an incompatible version of PyTorch and would prevent any possible installation of the two together). The end user might also be needing a more recent version of PyTorch.Given that PyTorch is almost always backward compatible, and quite stable, I think the requirements for it could be changed from
==1.8.1to>=1.8.1. I believe the same would be true for the other packages.