From 9a1cf4546c6d0e6accb17dbf3229351e531e0918 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Wed, 20 Nov 2019 19:05:28 +0800 Subject: [PATCH] Modify demo of sa nas. --- demo/nas/sa_nas_mobilenetv2_cifar10.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/nas/sa_nas_mobilenetv2_cifar10.py b/demo/nas/sa_nas_mobilenetv2_cifar10.py index 3e903960..249d4c21 100644 --- a/demo/nas/sa_nas_mobilenetv2_cifar10.py +++ b/demo/nas/sa_nas_mobilenetv2_cifar10.py @@ -39,7 +39,7 @@ def init_sa_nas(config): search_steps = 10000000 ### start a server and a client - sa_nas = SANAS(config, max_flops=base_flops, search_steps=search_steps) + sa_nas = SANAS(config, search_steps=search_steps, is_server=True) ### start a client, server_addr is server address #sa_nas = SANAS(config, max_flops = base_flops, server_addr=("10.255.125.38", 18607), search_steps = search_steps, is_server=False) -- GitLab