Underc0de - La Casa de los Informáticos

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
You are not allowed to view links. You are not allowed to view links. Register or Login or You are not allowed to view links. Register or Login
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!