Update hytra_supernet.py

use restricted paths by default
jefyjiang · Aug 31, 2021 · fa90337 · fa90337
1 parent 3e7018b
commit fa90337
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/searching/bossnas/models/supernets/hytra_supernet.py b/searching/bossnas/models/supernets/hytra_supernet.py
@@ -3,7 +3,7 @@
 import numpy as np
 import torch.nn as nn
 
-from bossnas.models.utils.hytra_paths import all_path
+from bossnas.models.utils.hytra_paths import all_path, restricted_path
 from bossnas.models.operations.operation_dict import OPS
 from bossnas.models.operations.operation_dict import reset
 from openselfsup.models.registry import BACKBONES
@@ -253,7 +253,8 @@ def set_forward_cfg(self, method='random', start_block=0): # support method: un
  return forward_op
 
  def get_all_path(self, start_block):
- return all_path[self.stage_depths[start_block]]
+ return restricted_path[self.stage_depths[start_block]]
+ # return all_path[self.stage_depths[start_block]]
 
  def reset_params(self):
  self.apply(reset)