modification

server.py

@@ -478,7 +478,7 @@ sorted_rank['merged'] = sorted_rank['merged'][llen * 3 // 4 : ]
 
 def generate_specific_attack_edge(start_entity, end_entity):
 
-    if device == 'cpu':
+    if not torch.cuda.is_available():
         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