NotFoundError (see above for traceback): Key local3/weights not found in checkpoint