iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update some relative path

Browse Source
This commit is contained in:
mhz 2024-07-29 00:21:41 +02:00
parent d470261a74
commit fb5e04988d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
graph_dit/naswot/naswot/datasets/data.py
View File

@ -1,5 +1,5 @@
from datasets import get_datasets from naswot.datasets import get_datasets
from config_utils import load_config from naswot.config_utils import load_config
import torch import torch
import torchvision import torchvision

2
graph_dit/naswot/naswot/utils.py
View File

@ -1,5 +1,5 @@
import torch import torch
from pycls.models.nas.nas import Cell from naswot.pycls.models.nas.nas import Cell
class DropChannel(torch.nn.Module): class DropChannel(torch.nn.Module):
def __init__(self, p, mod): def __init__(self, p, mod):