Have a problem, when using ‘https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_0.25_224/model.json’ model for the assignment the model trains and I can predict and save the model, but when I submit it flatten_Flatten1_input should have shape (7, 7, 256) but got array with shape (7, 7, 1024).
When I use the suggested model ‘https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_1.0_224/model.json’ the model in training gets an error WebGL:
‘’‘CONTEXT_LOST_WEBGL: loseContext: context lost
uncaught (in promise) Error: Failed to compile fragment shader.’‘’
and I can neither predict nor save the model for submission, what should I do in this case?
Basically the problem is the mobilenet model itself, because if I use another everything works perfectly fine, just the grader is unhappy!!!