def task()

in 2-dl-container/Container-Root/job/resnet/direct_benchmark-inf.py [0:0]

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

def task(model, cur_img_preprocess):
    global latency_list
    begin = time.time()

    batch_input_tensor = torch.cat([cur_img_preprocess] * batch_size)

    prediction = model(batch_input_tensor)
    latency_time = time.time() - begin

    latency_list.append(latency_time)
    return