I trained the model with two gpus and use
self.log('train_loss', loss) to log my loss in
There’re two losses every step which made it hard to read. Any solution to solve the problem?
I install lightning by
pip install git+https://github.com/PytorchLightning/pytorch-lightning.git@master --upgrade yesterday. I think it’s 0.10.0rc1.