Spaces:
Sleeping
Sleeping
File size: 2,298 Bytes
754afec fc5fa78 754afec fc5fa78 754afec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import argparse
import yaml
import logging
import logging.config
from pathlib import Path
from src.server.coordinator import FederatedCoordinator
from src.client.model import FederatedClient
def setup_logging(config):
"""Setup logging configuration."""
# Create logs directory if it doesn't exist
Path("logs").mkdir(exist_ok=True)
log_level = (config.get('monitoring', {}).get('log_level')
or config.get('server', {}).get('monitoring', {}).get('log_level')
or config.get('client', {}).get('monitoring', {}).get('log_level')
or 'INFO')
# Configure logging with UTF-8 encoding
logging.basicConfig(
level=log_level,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[
logging.StreamHandler(),
logging.FileHandler('logs/federated_learning.log', mode='a', encoding='utf-8')
]
)
# Reduce TensorFlow logging noise
logging.getLogger('tensorflow').setLevel(logging.WARNING)
# Create a divider in the log file
logger = logging.getLogger(__name__)
logger.info("\n" + "="*50)
logger.info("New Training Session Started")
logger.info("="*50 + "\n")
def load_config(config_path: str) -> dict:
with open(config_path, 'r') as f:
return yaml.safe_load(f)
def main():
parser = argparse.ArgumentParser(description='Federated Learning Demo')
parser.add_argument('--mode', choices=['server', 'client'], required=True)
parser.add_argument('--config', type=str, required=True)
args = parser.parse_args()
config = load_config(args.config)
setup_logging(config)
logger = logging.getLogger(__name__)
if args.mode == 'server':
coordinator = FederatedCoordinator(config)
logger.info("Starting federated server...")
coordinator.start()
else:
# Extract client ID from config or use default
client_id = config.get('client', {}).get('id', '1')
server_url = config.get('client', {}).get('server_url', 'http://localhost:8080')
client = FederatedClient(client_id, config, server_url)
logger.info(f"Starting federated client with ID: {client_id}")
client.start()
if __name__ == "__main__":
main() |