From fb5e04988d766e6cd1fc9394254b8a04a93f3183 Mon Sep 17 00:00:00 2001 From: mhz Date: Mon, 29 Jul 2024 00:21:41 +0200 Subject: [PATCH] update some relative path --- graph_dit/naswot/naswot/datasets/data.py | 4 ++-- graph_dit/naswot/naswot/utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/graph_dit/naswot/naswot/datasets/data.py b/graph_dit/naswot/naswot/datasets/data.py index e9d8959..aef3651 100644 --- a/graph_dit/naswot/naswot/datasets/data.py +++ b/graph_dit/naswot/naswot/datasets/data.py @@ -1,5 +1,5 @@ -from datasets import get_datasets -from config_utils import load_config +from naswot.datasets import get_datasets +from naswot.config_utils import load_config import torch import torchvision diff --git a/graph_dit/naswot/naswot/utils.py b/graph_dit/naswot/naswot/utils.py index 800f42f..ed8bed5 100644 --- a/graph_dit/naswot/naswot/utils.py +++ b/graph_dit/naswot/naswot/utils.py @@ -1,5 +1,5 @@ import torch -from pycls.models.nas.nas import Cell +from naswot.pycls.models.nas.nas import Cell class DropChannel(torch.nn.Module): def __init__(self, p, mod):