def print_times()

in dataflux_pytorch/benchmark/checkpointing/multinode/train.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

def print_times(args, avg_save_time, avg_load_time):
    avg_save_time_str = str(avg_save_time) + " seconds"
    avg_load_time_str = str(avg_load_time) + " seconds"
    if args.save_only:
        avg_load_time_str = "skipped"
    elif args.load_only:
        avg_save_time_str = "skipped"

    print("##################################")
    print("Average time to save one checkpoint: " + avg_save_time_str)
    print("Average time to load one checkpoint: " + avg_load_time_str)
    print("##################################")