diff --git a/PyTorch/dev/cv/image_classification/FixMatch_ID0989_for_PyTorch/train.py b/PyTorch/dev/cv/image_classification/FixMatch_ID0989_for_PyTorch/train.py index 3f8baad87a3284ff734421c64315708834d4155a..06c94f2dd1d58ab0c8723bd8f01d866286210671 100644 --- a/PyTorch/dev/cv/image_classification/FixMatch_ID0989_for_PyTorch/train.py +++ b/PyTorch/dev/cv/image_classification/FixMatch_ID0989_for_PyTorch/train.py @@ -54,6 +54,7 @@ from dataset.cifar import DATASET_GETTERS from utils import AverageMeter, accuracy import apex +torch.npu.set_start_fuzz_compile_step(3) logger = logging.getLogger(__name__) best_acc = 0 @@ -365,6 +366,8 @@ def train(args, labeled_trainloader, unlabeled_trainloader, test_loader, p_bar = tqdm(range(args.eval_step), disable=args.local_rank not in [-1, 0]) for batch_idx in range(args.eval_step): + #模糊编译 + torch.npu.global_step_inc() try: inputs_x, targets_x = labeled_iter.next() except: