mildnet
mildnet copied to clipboard
RuntimeError: Failed. Model function mildnet_vgg16 not found
I try to run your model !bash gcloud.local.run.keras.sh $MILDNET_CONFIG on my Jupyter but it has the RuntimeError: Failed. Model function mildnet_vgg16 not found
please kindly help to solve this problem.
Best Regards,
More details; job_configs/mildnet.cnf Python 2.7.16 TF 2.3.0
in task.py at line 8,9, I have to commented #from tensorflow import set_random_seed #set_random_seed(2) and changed to the import tensorflow tensorflow.random.set_seed(2)
and add @tf.function before def main