torch>=1.6
pytorch-lightning>=1.1.1, <1.2