Open giorg0s opened 4 years ago
Al acabar el proceso PyTorch -> ONNX -> TF y visualizar el grafo resultante con tensorboard lo que obtengo es esto (concretamente usando un modelo de mobilenet_v2 pre-entrenado): Creo que no tiene mucha pinta de haberlo hecho bien.
Parece algo raro, sí. Lo suyo sería mirar la arquitectura de red de mobilenet_v2 ya implementada en tensorflow y ver como es.
Parece algo raro, sí. Lo suyo sería mirar la arquitectura de red de mobilenet_v2 ya implementada en tensorflow y ver como es.
Sí, creo que el problema está cuando paso el modelo de ONNX a tensorflow, porque al visualizarlo tiene sentido:
Según comenta Andrés los pasos a seguir podrían ser:
La conversión idónea debería ser de Pytorch 1.6 a TF 1.14-15 o TF 2.