pse-ecn/pose-sensitive-embedding

Problems with training the Views Predictor

Closed this issue · 4 comments

zhang0931 commented

hi,

Thank the author, you job is so great. I successfully trained once, but there was a problem when I trained again .The problem is shown in the screenshot
I execute the following:
sudo python3 trainer_views.py --output=./output/Views_mode_new --data=/data/zhangli/datasets/RAP_dataset --dataset-name=rap --batch-size=16 --num-epochs=100 --network-name=resnet_v1_50_views --initial-checkpoint=/data/zhangli/pose-sensitive-embedding-master/checkpoint_dir/resnet_v1_50.ckpt --checkpoint-exclude-scopes=resnet_v1_50/logits --trainable-scopes=resnet_v1_50/3Views

image
image
image
image
image

andreas-eberle commented

Try running it with --ignore-missing-variables. Be aware that this randomly initializes the variables not available in the checkpoint you provide (the variables listed in the error message).

SpiritBear000 commented

hi,
I execute the following:
python3 trainer_views.py --output=everything_generate/output_dir --data=/media/lab210/Work/WH_HOME/SubjectDownload/Project/DATASETS/datasets/RAP/RAP_dataset --dataset-name=rap --batch-size=16 --num-epochs=100 --network-name=resnet_v1_50_views --initial-checkpoint=/media/lab210/Work/WH_HOME/SubjectDownload/Project/reid/pose-sensitive-embedding/everything_generate/checkpoint/resnet_v1_50.ckpt --checkpoint-exclude-scopes=everything_generate/resnet_v1_50/logits --trainable-scopes=resnet_v1_50/3Views --ignore-missing-variables

but get the error:
2018-10-02 12-30-43

The 'resnet_v1_50.ckpt' was downloaded from 'https://github.com/tensorflow/models/tree/master/research/slim'

And another question:

--checkpoint-exclude-scopes=everything_generate/resnet_v1_50/logits

where is the logits?

ssarfraz commented

try --checkpoint-exclude-scopes= resnet_v1_50/logits i.e. with only the network name, dont need to append the path

SpiritBear000 commented