torch~=1.6
pytorch-lightning~=1.1
