pytorch-lightning-smoke-detection2