yqzhishen
8784ed65d3
* [DONE]New AUX_Decoder/Backbone Network : LYNXNet (#200) * Update __init__.py * Update LYNXNet * add dropout * Lynxnet outnorm (#206) * post-norm * fix * add norm+mlp * Update LYNXNet.py * Update LYNXNetDecoder.py * do not need mlp * do not need mlp * Add out norm for LYNXNET * Add out norm for LYNXNETDecoder * delete lynxnet aux_decoder (#212) * refactor configuration options * fix onnx exporter for lynxnet * Add Pytorch version check when export onnx (#216) * recommended lynxnet hyperparameters for variance models * remove invalid items * Refactor code * Finish configuration schemas --------- Co-authored-by: KakaruHayate <97896816+KakaruHayate@users.noreply.github.com> Co-authored-by: yxlllc <llc1995@sina.com>
25 lines
750 B
Python
25 lines
750 B
Python
def get_backbone_type(root_config: dict, nested_config: dict = None):
|
|
if nested_config is None:
|
|
nested_config = root_config
|
|
return nested_config.get(
|
|
'backbone_type',
|
|
root_config.get(
|
|
'backbone_type',
|
|
root_config.get('diff_decoder_type', 'wavenet')
|
|
)
|
|
)
|
|
|
|
|
|
def get_backbone_args(config: dict, backbone_type: str):
|
|
args = config.get('backbone_args')
|
|
if args is not None:
|
|
return args
|
|
elif backbone_type == 'wavenet':
|
|
return {
|
|
'num_layers': config.get('residual_layers'),
|
|
'num_channels': config.get('residual_channels'),
|
|
'dilation_cycle_length': config.get('dilation_cycle_length'),
|
|
}
|
|
else:
|
|
return None
|