[QST] Help with end-to-end example of yoochoose

[PreyumKr]

❓ Questions & Help

Details

The end-to-end session based with yoochoose example notebook is getting stuck at the last training iteration on setting gpu size greater than 2. The code works fine on 1 and 2 gpus but on 3 and 4 GPU the iteration stuck at training output od 180th day...

### Tasks