From 1d4151da88ec6973fa1abbe08a9af3e116a93bd1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=B1=E4=BF=8A=E8=BE=89?= <11181965+zhujunhui1@user.noreply.gitee.com> Date: Tue, 11 Apr 2023 09:10:23 +0000 Subject: [PATCH] =?UTF-8?q?=E6=B3=A8=E9=87=8A=E9=BB=91=E5=90=8D=E5=8D=95?= =?UTF-8?q?=E7=AE=97=E5=AD=90update=20PyTorch/built-in/cv/classification/S?= =?UTF-8?q?hufflenetv2=5Ffor=5FPyTorch/8p=5Fmain=5Fmed.py.=20=E7=B2=BE?= =?UTF-8?q?=E5=BA=A6=EF=BC=8C=E6=80=A7=E8=83=BD=E9=AA=8C=E8=AF=81OK?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 朱俊辉 <11181965+zhujunhui1@user.noreply.gitee.com> --- .../classification/Shufflenetv2_for_PyTorch/8p_main_med.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/PyTorch/built-in/cv/classification/Shufflenetv2_for_PyTorch/8p_main_med.py b/PyTorch/built-in/cv/classification/Shufflenetv2_for_PyTorch/8p_main_med.py index e8df776c05..71c37457c7 100644 --- a/PyTorch/built-in/cv/classification/Shufflenetv2_for_PyTorch/8p_main_med.py +++ b/PyTorch/built-in/cv/classification/Shufflenetv2_for_PyTorch/8p_main_med.py @@ -611,7 +611,7 @@ def get_pytorch_val_loader(data_path, batch_size, workers=5, _worker_init_fn=Non if __name__ == '__main__': - option = {} - option["NPU_FUZZY_COMPILE_BLACKLIST"] = "BNTrainingUpdate" - torch.npu.set_option(option) + #option = {} + #option["NPU_FUZZY_COMPILE_BLACKLIST"] = "BNTrainingUpdate" + #torch.npu.set_option(option) main() -- Gitee