google-research / tf-slim

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can I do prediction by setting "is_training==False"

p1n0cch10 opened this issue · comments

Hi,

I am using the tf slim inception_v3 model. The flag is_training is set True when training.

I checked the code, and found that is_training is in arg_scope for the model.

I am confused that the calculating mechanics for training and prediction are different. I.e. the mean and the variance in batch normalization are calculated each batch when training, but the stored mean and variance are used when prediction.

How can I set the flags when prediction? Many thanks.