# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: protos/ensemble.proto
"""Generated protocol buffer code."""
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()


from protos import main_pb2 as protos_dot_main__pb2


DESCRIPTOR = _descriptor.FileDescriptor(
  name='protos/ensemble.proto',
  package='hierarchical_ood.protos',
  syntax='proto2',
  serialized_options=None,
  create_key=_descriptor._internal_create_key,
  serialized_pb=b'\n\x15protos/ensemble.proto\x12\x17hierarchical_ood.protos\x1a\x11protos/main.proto\"\xd7\t\n\x08\x45nsemble\x12?\n\x05model\x18\x01 \x01(\x0e\x32\'.hierarchical_ood.protos.Ensemble.Model:\x07SOFTMAX\x12\x1d\n\x08\x64\x61ta_dir\x18\x02 \x01(\t:\x0b\x64\x61ta/coarse\x12#\n\x0chierarchy_fn\x18\x03 \x01(\t:\rpruned-wn.pth\x12:\n\x0ctrain_params\x18\x04 \x01(\x0b\x32$.hierarchical_ood.protos.TrainParams\x12\x31\n\x03sgd\x18\x05 \x01(\x0b\x32\".hierarchical_ood.protos.SGDParamsH\x00\x12\x33\n\x04\x61\x64\x61m\x18\x06 \x01(\x0b\x32#.hierarchical_ood.protos.AdamParamsH\x00\x12\x33\n\x02\x63\x65\x18\x07 \x01(\x0b\x32%.hierarchical_ood.protos.CrossEntropyH\x01\x12:\n\x03\x62\x63\x65\x18\x08 \x01(\x0b\x32+.hierarchical_ood.protos.BinaryCrossEntropyH\x01\x12\x37\n\x02hl\x18\t \x01(\x0b\x32).hierarchical_ood.protos.HierarchicalLossH\x01\x12.\n\x02ml\x18\n \x01(\x0b\x32 .hierarchical_ood.protos.MOSLossH\x01\x12\x0c\n\x04seed\x18\x0b \x03(\x05\x12\x15\n\nnum_models\x18\x0c \x01(\x05:\x01\x35\x12\x17\n\x0crepeat_iters\x18\x1a \x01(\x05:\x01\x31\x12\x16\n\x07no_save\x18\r \x01(\x08:\x05\x66\x61lse\x12\x16\n\x07verbose\x18\x0e \x01(\x08:\x05\x66\x61lse\x12\x11\n\x07savedir\x18\x0f \x01(\t:\x00\x12\x1a\n\x08\x62\x61\x63kbone\x18\x10 \x01(\t:\x08resnet50\x12\x36\n\nsoftmax_mc\x18\x11 \x01(\x0b\x32 .hierarchical_ood.protos.SoftmaxH\x02\x12\x36\n\ncascade_mc\x18\x12 \x01(\x0b\x32 .hierarchical_ood.protos.CascadeH\x02\x12\x42\n\x10\x63\x61scadefchead_mc\x18\x13 \x01(\x0b\x32&.hierarchical_ood.protos.CascadeFCHeadH\x02\x12\x42\n\x10softmaxfchead_mc\x18\x14 \x01(\x0b\x32&.hierarchical_ood.protos.SoftmaxFCHeadH\x02\x12\x32\n\x06mos_mc\x18\x15 \x01(\x0b\x32 .hierarchical_ood.protos.MOSConfH\x02\x12\x15\n\rfar_ood_dsets\x18\x16 \x03(\t\x12\x1f\n\x15\x64istribution_strategy\x18\x17 \x01(\t:\x00\x12\x1f\n\x10resume_from_ckpt\x18\x18 \x01(\x08:\x05\x66\x61lse\x12\x1c\n\x12\x66inetune_from_ckpt\x18\x19 \x01(\t:\x00\"c\n\x05Model\x12\x0b\n\x07SOFTMAX\x10\x01\x12\x07\n\x03ILR\x10\x02\x12\x08\n\x04HILR\x10\x03\x12\x0b\n\x07\x43\x41SCADE\x10\x04\x12\x11\n\rCASCADEFCHEAD\x10\x05\x12\x11\n\rSOFTMAXFCHEAD\x10\x06\x12\x07\n\x03MOS\x10\x07\x42\x0b\n\toptimizerB\x06\n\x04lossB\x0e\n\x0cmodel_config'
  ,
  dependencies=[protos_dot_main__pb2.DESCRIPTOR,])



_ENSEMBLE_MODEL = _descriptor.EnumDescriptor(
  name='Model',
  full_name='hierarchical_ood.protos.Ensemble.Model',
  filename=None,
  file=DESCRIPTOR,
  create_key=_descriptor._internal_create_key,
  values=[
    _descriptor.EnumValueDescriptor(
      name='SOFTMAX', index=0, number=1,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='ILR', index=1, number=2,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='HILR', index=2, number=3,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='CASCADE', index=3, number=4,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='CASCADEFCHEAD', index=4, number=5,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='SOFTMAXFCHEAD', index=5, number=6,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
    _descriptor.EnumValueDescriptor(
      name='MOS', index=6, number=7,
      serialized_options=None,
      type=None,
      create_key=_descriptor._internal_create_key),
  ],
  containing_type=None,
  serialized_options=None,
  serialized_start=1173,
  serialized_end=1272,
)
_sym_db.RegisterEnumDescriptor(_ENSEMBLE_MODEL)


_ENSEMBLE = _descriptor.Descriptor(
  name='Ensemble',
  full_name='hierarchical_ood.protos.Ensemble',
  filename=None,
  file=DESCRIPTOR,
  containing_type=None,
  create_key=_descriptor._internal_create_key,
  fields=[
    _descriptor.FieldDescriptor(
      name='model', full_name='hierarchical_ood.protos.Ensemble.model', index=0,
      number=1, type=14, cpp_type=8, label=1,
      has_default_value=True, default_value=1,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='data_dir', full_name='hierarchical_ood.protos.Ensemble.data_dir', index=1,
      number=2, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"data/coarse".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='hierarchy_fn', full_name='hierarchical_ood.protos.Ensemble.hierarchy_fn', index=2,
      number=3, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"pruned-wn.pth".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='train_params', full_name='hierarchical_ood.protos.Ensemble.train_params', index=3,
      number=4, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='sgd', full_name='hierarchical_ood.protos.Ensemble.sgd', index=4,
      number=5, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='adam', full_name='hierarchical_ood.protos.Ensemble.adam', index=5,
      number=6, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='ce', full_name='hierarchical_ood.protos.Ensemble.ce', index=6,
      number=7, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='bce', full_name='hierarchical_ood.protos.Ensemble.bce', index=7,
      number=8, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='hl', full_name='hierarchical_ood.protos.Ensemble.hl', index=8,
      number=9, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='ml', full_name='hierarchical_ood.protos.Ensemble.ml', index=9,
      number=10, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='seed', full_name='hierarchical_ood.protos.Ensemble.seed', index=10,
      number=11, type=5, cpp_type=1, label=3,
      has_default_value=False, default_value=[],
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='num_models', full_name='hierarchical_ood.protos.Ensemble.num_models', index=11,
      number=12, type=5, cpp_type=1, label=1,
      has_default_value=True, default_value=5,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='repeat_iters', full_name='hierarchical_ood.protos.Ensemble.repeat_iters', index=12,
      number=26, type=5, cpp_type=1, label=1,
      has_default_value=True, default_value=1,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='no_save', full_name='hierarchical_ood.protos.Ensemble.no_save', index=13,
      number=13, type=8, cpp_type=7, label=1,
      has_default_value=True, default_value=False,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='verbose', full_name='hierarchical_ood.protos.Ensemble.verbose', index=14,
      number=14, type=8, cpp_type=7, label=1,
      has_default_value=True, default_value=False,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='savedir', full_name='hierarchical_ood.protos.Ensemble.savedir', index=15,
      number=15, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='backbone', full_name='hierarchical_ood.protos.Ensemble.backbone', index=16,
      number=16, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"resnet50".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='softmax_mc', full_name='hierarchical_ood.protos.Ensemble.softmax_mc', index=17,
      number=17, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='cascade_mc', full_name='hierarchical_ood.protos.Ensemble.cascade_mc', index=18,
      number=18, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='cascadefchead_mc', full_name='hierarchical_ood.protos.Ensemble.cascadefchead_mc', index=19,
      number=19, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='softmaxfchead_mc', full_name='hierarchical_ood.protos.Ensemble.softmaxfchead_mc', index=20,
      number=20, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='mos_mc', full_name='hierarchical_ood.protos.Ensemble.mos_mc', index=21,
      number=21, type=11, cpp_type=10, label=1,
      has_default_value=False, default_value=None,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='far_ood_dsets', full_name='hierarchical_ood.protos.Ensemble.far_ood_dsets', index=22,
      number=22, type=9, cpp_type=9, label=3,
      has_default_value=False, default_value=[],
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='distribution_strategy', full_name='hierarchical_ood.protos.Ensemble.distribution_strategy', index=23,
      number=23, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='resume_from_ckpt', full_name='hierarchical_ood.protos.Ensemble.resume_from_ckpt', index=24,
      number=24, type=8, cpp_type=7, label=1,
      has_default_value=True, default_value=False,
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
    _descriptor.FieldDescriptor(
      name='finetune_from_ckpt', full_name='hierarchical_ood.protos.Ensemble.finetune_from_ckpt', index=25,
      number=25, type=9, cpp_type=9, label=1,
      has_default_value=True, default_value=b"".decode('utf-8'),
      message_type=None, enum_type=None, containing_type=None,
      is_extension=False, extension_scope=None,
      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
  ],
  extensions=[
  ],
  nested_types=[],
  enum_types=[
    _ENSEMBLE_MODEL,
  ],
  serialized_options=None,
  is_extendable=False,
  syntax='proto2',
  extension_ranges=[],
  oneofs=[
    _descriptor.OneofDescriptor(
      name='optimizer', full_name='hierarchical_ood.protos.Ensemble.optimizer',
      index=0, containing_type=None,
      create_key=_descriptor._internal_create_key,
    fields=[]),
    _descriptor.OneofDescriptor(
      name='loss', full_name='hierarchical_ood.protos.Ensemble.loss',
      index=1, containing_type=None,
      create_key=_descriptor._internal_create_key,
    fields=[]),
    _descriptor.OneofDescriptor(
      name='model_config', full_name='hierarchical_ood.protos.Ensemble.model_config',
      index=2, containing_type=None,
      create_key=_descriptor._internal_create_key,
    fields=[]),
  ],
  serialized_start=70,
  serialized_end=1309,
)

_ENSEMBLE.fields_by_name['model'].enum_type = _ENSEMBLE_MODEL
_ENSEMBLE.fields_by_name['train_params'].message_type = protos_dot_main__pb2._TRAINPARAMS
_ENSEMBLE.fields_by_name['sgd'].message_type = protos_dot_main__pb2._SGDPARAMS
_ENSEMBLE.fields_by_name['adam'].message_type = protos_dot_main__pb2._ADAMPARAMS
_ENSEMBLE.fields_by_name['ce'].message_type = protos_dot_main__pb2._CROSSENTROPY
_ENSEMBLE.fields_by_name['bce'].message_type = protos_dot_main__pb2._BINARYCROSSENTROPY
_ENSEMBLE.fields_by_name['hl'].message_type = protos_dot_main__pb2._HIERARCHICALLOSS
_ENSEMBLE.fields_by_name['ml'].message_type = protos_dot_main__pb2._MOSLOSS
_ENSEMBLE.fields_by_name['softmax_mc'].message_type = protos_dot_main__pb2._SOFTMAX
_ENSEMBLE.fields_by_name['cascade_mc'].message_type = protos_dot_main__pb2._CASCADE
_ENSEMBLE.fields_by_name['cascadefchead_mc'].message_type = protos_dot_main__pb2._CASCADEFCHEAD
_ENSEMBLE.fields_by_name['softmaxfchead_mc'].message_type = protos_dot_main__pb2._SOFTMAXFCHEAD
_ENSEMBLE.fields_by_name['mos_mc'].message_type = protos_dot_main__pb2._MOSCONF
_ENSEMBLE_MODEL.containing_type = _ENSEMBLE
_ENSEMBLE.oneofs_by_name['optimizer'].fields.append(
  _ENSEMBLE.fields_by_name['sgd'])
_ENSEMBLE.fields_by_name['sgd'].containing_oneof = _ENSEMBLE.oneofs_by_name['optimizer']
_ENSEMBLE.oneofs_by_name['optimizer'].fields.append(
  _ENSEMBLE.fields_by_name['adam'])
_ENSEMBLE.fields_by_name['adam'].containing_oneof = _ENSEMBLE.oneofs_by_name['optimizer']
_ENSEMBLE.oneofs_by_name['loss'].fields.append(
  _ENSEMBLE.fields_by_name['ce'])
_ENSEMBLE.fields_by_name['ce'].containing_oneof = _ENSEMBLE.oneofs_by_name['loss']
_ENSEMBLE.oneofs_by_name['loss'].fields.append(
  _ENSEMBLE.fields_by_name['bce'])
_ENSEMBLE.fields_by_name['bce'].containing_oneof = _ENSEMBLE.oneofs_by_name['loss']
_ENSEMBLE.oneofs_by_name['loss'].fields.append(
  _ENSEMBLE.fields_by_name['hl'])
_ENSEMBLE.fields_by_name['hl'].containing_oneof = _ENSEMBLE.oneofs_by_name['loss']
_ENSEMBLE.oneofs_by_name['loss'].fields.append(
  _ENSEMBLE.fields_by_name['ml'])
_ENSEMBLE.fields_by_name['ml'].containing_oneof = _ENSEMBLE.oneofs_by_name['loss']
_ENSEMBLE.oneofs_by_name['model_config'].fields.append(
  _ENSEMBLE.fields_by_name['softmax_mc'])
_ENSEMBLE.fields_by_name['softmax_mc'].containing_oneof = _ENSEMBLE.oneofs_by_name['model_config']
_ENSEMBLE.oneofs_by_name['model_config'].fields.append(
  _ENSEMBLE.fields_by_name['cascade_mc'])
_ENSEMBLE.fields_by_name['cascade_mc'].containing_oneof = _ENSEMBLE.oneofs_by_name['model_config']
_ENSEMBLE.oneofs_by_name['model_config'].fields.append(
  _ENSEMBLE.fields_by_name['cascadefchead_mc'])
_ENSEMBLE.fields_by_name['cascadefchead_mc'].containing_oneof = _ENSEMBLE.oneofs_by_name['model_config']
_ENSEMBLE.oneofs_by_name['model_config'].fields.append(
  _ENSEMBLE.fields_by_name['softmaxfchead_mc'])
_ENSEMBLE.fields_by_name['softmaxfchead_mc'].containing_oneof = _ENSEMBLE.oneofs_by_name['model_config']
_ENSEMBLE.oneofs_by_name['model_config'].fields.append(
  _ENSEMBLE.fields_by_name['mos_mc'])
_ENSEMBLE.fields_by_name['mos_mc'].containing_oneof = _ENSEMBLE.oneofs_by_name['model_config']
DESCRIPTOR.message_types_by_name['Ensemble'] = _ENSEMBLE
_sym_db.RegisterFileDescriptor(DESCRIPTOR)

Ensemble = _reflection.GeneratedProtocolMessageType('Ensemble', (_message.Message,), {
  'DESCRIPTOR' : _ENSEMBLE,
  '__module__' : 'protos.ensemble_pb2'
  # @@protoc_insertion_point(class_scope:hierarchical_ood.protos.Ensemble)
  })
_sym_db.RegisterMessage(Ensemble)


# @@protoc_insertion_point(module_scope)
