I am unable to load parameters from VGG model.
Below is the code.
"tf.random.set_seed(272) # DO NOT CHANGE THIS VALUE
pp = pprint.PrettyPrinter(indent=4)
img_size = 400
vgg = tf.keras.applications.VGG19(include_top=False,
input_shape=(img_size, img_size, 3),
weights=‘pretrained-model/vgg19_weights_tf_dim_ordering_tf_kernels_notop.h5’)
vgg.trainable = False
pp.pprint(vgg)"
I am getting the following error.
“Internal Error: CUDA runtime implicit initialization on GPU:0 failed. Status: out of memory”