bitorch.models.dlrm.parse_layer_sizes

bitorch.models.dlrm.parse_layer_sizes(layer_sizes_str: Union[List[int], str]) → List[int]

parses layer sizes passed as string via cli arg

Parameters:

• layer_sizes_str (Union[List[int], str]) – either list of layer sizes in which case the input is just returned or

• a (a string in format '[layer size) –

• b (layer size) –

• etc]' –

Returns:

list of layer sizes

Return type:

List[int]