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

# ❓ 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...

```[tasklist]
### Tasks
```
