diff --git a/model.py b/model.py index c5e9ceb..c780119 100644 --- a/model.py +++ b/model.py @@ -62,7 +62,7 @@ def model_fn(features, labels, mode, params=None): # Finetune just supervised (linear) head will not update BN stats. model_train_mode = False else: - # Pretrain or finetuen anything else will update BN stats. + # Pretrain or finetune anything else will update BN stats. model_train_mode = is_training hiddens = model(features, is_training=model_train_mode)