Google Coolab Crashes

While trying to use the existing model the program crashes. Is there any way to solve this? The code where it crashes is:

number of GPUs to use

num_gpus = 1

ctx = [mx.gpu(i) for i in range(num_gpus)]

Get the model CIFAR_ResNet20_v1, with 10 output classes, without pre-trained weights

net = get_model(‘cifar_resnet20_v1’, classes=10) (this line)

net.initialize(mx.init.Xavier(), ctx = ctx)

Warning says: WARNING:root:kernel 478b9c05-09b9-49c9-9834-226a60be6c08 restarted

Would be grateful if people can help me out

Did you select GPU as Hardware accelerator?