def __call__(self, model):
"""Run constructor."""
Slot.reset()
ParamSpace().reset()
seed = self.seed
if seed:
init_device(self.device, seed)
configure_ops(self.ops_conf or {})
return model
def __call__(self, model):
"""Run Exporter."""
Slot.set_export_fn(self.export)
arch_desc = []
gen = self.gen or Slot.gen_slots_model(model)
for m in gen():
if m in self.visited:
continue
arch_desc.append(m.to_arch_desc(**copy.deepcopy(self.fn_args)))
self.visited.clear()
return arch_desc
def __call__(self, model):
"""Run constructor."""
gen = self.gen or Slot.gen_slots_model(model)
all_slots = list(gen())
for m in all_slots:
if self.skip_exist and m.get_entity() is not None:
continue
ent = self.convert(m)
if ent is not None:
m.set_entity(ent)
return model
def __call__(self, model):
"""Run constructor."""
Slot.set_convert_fn(self.convert)
return self.visit(model)
def __call__(self, model):
"""Run Exporter."""
Slot.set_export_fn(self.export)
desc = self.visit(model)
self.visited.clear()
return desc