JaminFong/DenseNAS

net_config

JacobLau84 opened this issue · 1 comments

JacobLau84 commented

您好，想请教一下我想要retrain自己的model的话，net_config这个档案要如何生成呢？

感谢。

JaminFong commented

您好，如果您的model与DenseNAS model类似（基于MobileNet或ResNet模块等），可以参照提供的net_config自行编写；如若不太合适编写类似的net_config，则可重新定义retrain.py中的model变量。

DenseNAS/run_apis/retrain.py

Line 65 in 5c3ecbb

model = derivedNetwork(config.net_config, config=config)

  • ❤️1