modification

server.py

@@ -478,6 +478,9 @@ sorted_rank['merged'] = sorted_rank['merged'][llen * 3 // 4 : ]
 
 def generate_specific_attack_edge(start_entity, end_entity):
 
+    if device == 'cpu':
+        print('We can just set the malicious link equals to the target link, since the generation of malicious link is too slow on cpu')
+        return start_entity, '10', end_entity
     global specific_model
         
     specific_model.to(device)