1// Copyright 2016 The TensorFlow Authors. All Rights Reserved. 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14// ============================================================================= 15 16syntax = "proto3"; 17 18package tensorflow; 19option cc_enable_arenas = true; 20 21// Protocol buffer holding hyper parameters. 22// Examples of hyper parameters: 23// learning_rate = 0.1, 24// num_hidden_units = 100, 25// activations = ['relu', 'tanh'] 26message HParamDef { 27 message BytesList { 28 repeated bytes value = 1; 29 } 30 message FloatList { 31 repeated float value = 1 [packed = true]; 32 } 33 message Int64List { 34 repeated int64 value = 1 [packed = true]; 35 } 36 message BoolList { 37 repeated bool value = 1 [packed = true]; 38 } 39 message HParamType { 40 oneof kind { 41 int64 int64_value = 1; 42 float float_value = 2; 43 bytes bytes_value = 3; 44 bool bool_value = 7; 45 Int64List int64_list = 4; 46 FloatList float_list = 5; 47 BytesList bytes_list = 6; 48 BoolList bool_list = 8; 49 } 50 }; 51 map<string, HParamType> hparam = 1; 52} 53