fix small bugs in DARTS-V1 for NASNet-Space

2020-01-17 22:28:33 +11:00
parent db2760c260
commit c2ff845d1b
3 changed files with 13 additions and 3 deletions
--- a/lib/models/cell_searchs/search_model_darts.py
+++ b/lib/models/cell_searchs/search_model_darts.py
@@ -53,6 +53,10 @@ class TinyNetworkDarts(nn.Module):
  def get_alphas(self):
    return [self.arch_parameters]

+  def show_alphas(self):
+    with torch.no_grad():
+      return 'arch-parameters :\n{:}'.format( nn.functional.softmax(self.arch_parameters, dim=-1).cpu() )
+
  def get_message(self):
    string = self.extra_repr()
    for i, cell in enumerate(self.cells):