def init_model()

in src/snn_fine_tune.py [0:0]

• 19 lines of code
• 1 McCabe index (conditional complexity)

def init_model(
    device,
    training,
    use_fp16,
    r_enc_path,
    iterations_per_epoch,
    world_size,
    start_lr,
    ref_lr,
    num_epochs,
    output_dim=128,
    model_name='resnet50',
    warmup_epochs=0,
    use_pred_head=False,
    use_lars=False,
    wd=1e-6,
    final_lr=0.,
    momentum=0.9,
    nesterov=False