Reputation: 85
System information
Describe the problem:
Initially, the problem I had was is described here: https://github.com/tensorflow/models/issues/2653
The solution given in that is to try and install tf nightly. I did that as shown : tfnightly
However, now when I try to run the following command:
C:\Users\kannan\Documents\MDLearning\models\research\object_detection>train.py --logtostderr --train_dir=training/ --
pipeline_config_path=training/ssd_mobilenet_v1_pets.config
I get the following Error:
FATAL Flags parsing error: Unknown command line flag 'logtostderr' Pass --helpshort or --helpfull to see help on flags.
Any idea how to solve this?
EDIT: I omitted --logtostderr and tried the same command. I probably ended up with a memory allocation error from the looks of it as seen here in the PDF attached below. I know it is really long. To make it easy, I have highlighted the important errors: Page 1, 2, 106, 107, 111 and 115.
Any idea how to get around this? I am using a batch size of 1 and it possibly can't go any lower. I have used around 187 images with two classes labelled in each image. All my images are 960 by 720 pixels each. My GPU compute capability is 3.0
Upvotes: 3
Views: 9630