Description
Was trying to run inference with the pre-trained model on Google Colab and ran into the following error:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation discriminator/discriminator_unit/input_stage/conv/Conv/weights/Initializer/random_uniform/RandomUniform: Could not satisfy explicit device specification '' because the node node discriminator/discriminator_unit/input_stage/conv/Conv/weights/Initializer/random_uniform/RandomUniform (defined at /content/gdrive/My Drive/Portrait_GAN_master/Portrait_GAN/SRGAN-tensorflow/lib/ops.py:40) placed on device Device assignments active during op 'discriminator/discriminator_unit/input_stage/conv/Conv/weights/Initializer/random_uniform/RandomUniform' creation:
I think it's a problem with running on Colab but I'm not sure. I don't really have any alternatives right now so I'd appreciate if you could help out.
Thanks!