Key in_hidden/batch_normalization/beta not found in checkpoint