update NAS-Bench

2020-03-09 19:38:00 +11:00
parent 9a83814a46
commit e59eb804cb
35 changed files with 693 additions and 64 deletions
--- a/lib/models/shape_infers/InferCifarResNet.py
+++ b/lib/models/shape_infers/InferCifarResNet.py
@@ -1,8 +1,9 @@
-import math, torch
+#####################################################
+# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019.01 #
+#####################################################
 import torch.nn as nn
 import torch.nn.functional as F
 from ..initialization import initialize_resnet
-from ..SharedUtils    import additive_func


 class ConvBNReLU(nn.Module):