Update base.py

Author: MatZaharia

golf_federated/client/process/config/device/base.py

@@ -148,4 +148,7 @@ def stop(self) -> None:
 
         # Call the trainer to stop local training.
         loggerhear.log("Client Info  ", "Stop training on %s!" % self.client_name)
-        self.trainer.stop()
+        try:
+            self.trainer.stop()
+        except:
+            self.trainer.stop_trainer()