GPU getting crashed

[Adwait01-Bhosale]

Sir after building model_5(tribrid model) of the Skimlit application project it's showing me that your system has been crashed (GPU).

Is there any solution to these problem.

[lukbast]

It depends where you run your experiments. Usually I train my model locally on my laptop with 6GB 1660 Ti. But when I tried to compile tribrid model, I was running out of memory. When I moved to colab, model compiled wihout any issues on K80.
So, if you also run experiments on your own machine try to move into colab. 12GB GPUs on Colab is enough for this model.

[Adwait01-Bhosale]

I got this error on colab itself

[kp9z]

I got the same error! I upgraded to Colab pro but it doesn't help, try to reduce batch size to 2, it doesn't help either

[kp9z]

A quick update: Instead of writing the full 5 model like on the video, I start fresh, with the final experiment (model_5: tribird), it works.
I guess I have introduced an error along the way when doing a lot of experiments

[mrdbourke]

Thank you for the update @kp9z, glad you found a fix

[Adwait01-Bhosale]

Thanks a lot @kp9z and @mrdbourke sir.