ffe5898317c4f971483f649a1ffb53d81a29705a,apps/kg/eval.py,,main,#Any#,92

Before Change

        total_metrics = {}
        for i in range(args.num_proc):
            metrics = queue.get()
            for k, v in metrics.items():
                if i == 0:
                    total_metrics[k] = v / args.num_proc
                else:
                    total_metrics[k] += v / args.num_proc
        for k, v in metrics.items():
            print("Test average {} at [{}/{}]: {}".format(k, args.step, args.max_step, v))
    else:
        test(args, model, [test_sampler_head, test_sampler_tail])

After Change

            logs = logs + log

        for metric in logs[0].keys():
            metrics[metric] = sum([log[metric] for log in logs]) / len(logs)
        for k, v in metrics.items():
            print("Test average {} at [{}/{}]: {}".format(k, args.step, args.max_step, v))

        for proc in procs:

Instances