INNER CODE UNIT · Python

do_eval_flag

OpenMeshLab/MeshXL · engine.py:142

            do_eval_flag = (curr_iter + 1) % args.eval_every_iteration == 0
            do_eval_flag &= (curr_iter + 1) > args.start_eval_after
            do_eval_flag |= (curr_iter + 1) == max_iters
            
            if do_eval_flag is True:
                eval_metrics = {}
                model.eval()
                with accelerator.autocast():
                    for test_loader in dataloaders['test']:
                        task_metrics, eval_loss_dict = test_loader.dataset.eval_func(
                            args,
                            curr_epoch,
                            accelerator.unwrap_model(model),
                            accelerator,
                            test_loader,
                            logger,
                            curr_train_iter=curr_iter
                        )

View source record →