Saltar al contenido
Underc0de /foro

[SOLUCIONADO] Google Colab / Tensorflow training Redes neuronales

Iniciado por LKI, Octubre 22, 2019, 08:02:17 PM

Tema anterior - Siguiente tema

0 Miembros y 1 Visitante están viendo este tema.

Octubre 22, 2019, 08:02:17 PM Ultima modificación: Octubre 25, 2019, 01:04:44 AM por Gabriela
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:

Código: text
"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.
Ser mejor cada día es mi meta

=LKI=

Para verlo mejor, deberías dejar el codigo del .py

Saludos,
ANTRAX


Cita de: ANTRAX en Octubre 22, 2019, 10:48:43 PM
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!
Ser mejor cada día es mi meta

=LKI=

Grupo de WhatsApp