Kenenbek R

I have this error when I try to launch a `custom-notebook-image` from [examples folder](https://github.com/jupyterhub/jupyterhub-deploy-docker/tree/master/examples/custom-notebook-server). ```[W 2019-12-23 19:14:00.701 JupyterHub web:1667] 500 GET /hub/user/kenenbek/ (::ffff:92.242.58.11): Spawner failed to start [status=ExitCode=1, Error='', FinishedAt=2019-12-23T19:13:51.834730009Z]....

The default image is working but the error happens when I change the environmental variable `DOCKER_NOTEBOOK_IMAGE` and `.env` file to the custom one.

If I define `SelfAttention` class with `reference_code` argument : ``` attention = trax.layers.SelfAttention(n_heads=2, , use_reference_code=True) ``` Everything works fine. Is it a **bug**?

Goland 2018.2 Ubuntu 16.04  The same problem — fix doesn't work.

It would be very nice to have this feature 🚀

Is this Warning still not solved?

@clement-pages It is unbelievable, before it was: ``` ✓ TEST COMPLETE! ⏱ Time taken: 17.57 minutes (1053.9 seconds) ``` Now it is: ``` ✓ TEST COMPLETE! ⏱ Time taken: 0.83...

@crysingzz here is full code ```python def diarize_audio(audio_file, num_speakers=None, device=None): # Set device if device is None: device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print(f"Using device: {device}") # Load pipeline...