lm_default_config = {
    ### GENERAL
    'seed': None,
    'verbose': True,
    'show_plots': True,
    
    ### TRAIN
    'train_config': {
        # Classifier
        # Class balance (if learn_class_balance=False, fix to class_balance)
        'learn_class_balance': False,
        # Class balance initialization / prior
        'class_balance_init': None, # (array) If None, assume uniform
        # Model params initialization / priors
        'mu_init': 0.4, 
        # L2 regularization (around prior values)
        'l2': 0.01,
        
        # Optimizer
        'optimizer_config': {
            'optimizer': 'sgd',
            'optimizer_common': {
                'lr': 0.01,
            },
            # Optimizer - SGD
            'sgd_config': {
                'momentum': 0.9, 
            },
        },

        # Scheduler
        'scheduler_config': {
            'scheduler': None, 
        },

        # Checkpointer
        'checkpoint': False,

        # Train loop
        'n_epochs': 100, 
        'print_every': 10, 
    },
}

lm_default_config = {
    ### GENERAL
    'seed': None,
    'verbose': True,
    'show_plots': True,
    
    ### TRAIN
    'train_config': {
        # Classifier
        # Class balance (if learn_class_balance=False, fix to class_balance)
        'learn_class_balance': False,
        # Class balance initialization / prior
        'class_balance_init': None, # (array) If None, assume uniform
        # Model params initialization / priors
        'mu_init': 0.4, 
        
        # Optimizer
        'optimizer_config': {
            'optimizer': 'sgd',
            'optimizer_common': {
                'lr': 0.01,
            },
            # Optimizer - SGD
            'sgd_config': {
                'momentum': 0.9, 
            },
        },

        # Scheduler
        'scheduler_config': {
            'scheduler': None, 
        },

        # Checkpointer
        'checkpoint': False,

        # Train loop
        'n_epochs': 100, 
        'print_every': 10, 
    },
}

Remove l2 from lm_default_config since it is currently unused