Refine lib -> xautodl

2021-05-19 07:19:20 +00:00
parent bda202ce87
commit 5b9a028e60
19 changed files with 46 additions and 46 deletions
--- a/xautodl/models/cell_infers/cells.py
+++ b/xautodl/models/cell_infers/cells.py
@@ -6,7 +6,7 @@ import torch
 import torch.nn as nn
 from copy import deepcopy

-from models.cell_operations import OPS
+from xautodl.models.cell_operations import OPS


 # Cell for NAS-Bench-201
--- a/xautodl/models/cell_infers/nasnet_cifar.py
+++ b/xautodl/models/cell_infers/nasnet_cifar.py
@@ -4,6 +4,7 @@
 import torch
 import torch.nn as nn
 from copy import deepcopy
+
 from .cells import NASNetInferCell as InferCell, AuxiliaryHeadCIFAR