To answer issue #119

2022-03-20 23:12:12 -07:00
parent d2cef525f3
commit 8d0799dfb1
3 changed files with 35 additions and 4 deletions
--- a/xautodl/models/cell_searchs/generic_model.py
+++ b/xautodl/models/cell_searchs/generic_model.py
@@ -347,6 +347,10 @@ class GenericNAS201Model(nn.Module):
                    feature = cell.forward_gdas(feature, alphas, index)
                    if self.verbose:
                        verbose_str += "-forward_gdas"
+                elif self.mode == "gdas_v1":
+                    feature = cell.forward_gdas_v1(feature, alphas, index)
+                    if self.verbose:
+                        verbose_str += "-forward_gdas_v1"
                else:
                    raise ValueError("invalid mode={:}".format(self.mode))
            else: