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