Fix small bugs

2019-04-08 11:04:08 +08:00
parent 3b1d8f1e4b
commit 36bb07ef1a
4 changed files with 13 additions and 6 deletions
--- a/lib/nas/ImageNet.py
+++ b/lib/nas/ImageNet.py
@@ -80,6 +80,9 @@ class NetworkImageNet(nn.Module):
  def update_drop_path(self, drop_path_prob):
    self.drop_path_prob = drop_path_prob

+  def get_drop_path(self):
+    return self.drop_path_prob
+
  def auxiliary_param(self):
    if self.auxiliary_head is None: return []
    else: return list( self.auxiliary_head.parameters() )