Underc0de

Foros Generales => Dudas y pedidos generales => Mensaje iniciado por: LKI en Octubre 22, 2019, 08:02:17 PM

Título: [SOLUCIONADO] Google Colab / Tensorflow training Redes neuronales
Publicado por: LKI en Octubre 22, 2019, 08:02:17 PM
Hola chicos! estoy empezando a usar Google Colab y tensorflow, estoy entrenando un modelo en particular "faster_rcnn_inception_v2_coco" con el siguiente comando:

"python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_coco.config

El entrenamiento se ejecuta correctamente pero los checkpoints no se guardan en la carpeta especificada, es decir, "training", por el contrario veo que se guardan en /tmp/tmpXXX y en cada nuevo entrenamiento se crea una nueva carpeta temp y no se resume desde el último checkpoint sino que empieza desde 0. ¿Alguien sabe como puedo solucionar esto? Saludos.
Título: Re:Google Colab / Tensorflow training Redes neuronales
Publicado por: ANTRAX en Octubre 22, 2019, 10:48:43 PM
Para verlo mejor, deberías dejar el codigo del .py

Saludos,
ANTRAX
Título: Re:Google Colab / Tensorflow training Redes neuronales
Publicado por: LKI en Octubre 24, 2019, 01:29:19 AM
No tienes permitido ver enlaces. Registrate o Entra a tu cuenta
Para verlo mejor, deberías dejar el codigo del .py

Saludos,
ANTRAX


Hola Antrax, gracias por tu respuesta, en realidad al estar usando una versión antigua del framework el comando training_dir no es reconocido, debería ser "model_dir". Lo raro es que el framework no te lo sugiera o avise. De todos modos, ya está solucionado. Gracias!