Skip to content

Commit

Permalink
laksdcn
Browse files Browse the repository at this point in the history
  • Loading branch information
TheColdIce committed Sep 29, 2023
1 parent 85a7e93 commit 273f791
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions federated-learning-v2/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def run_experiment(name, n_workers, trainsets, testsets, Module, Optimizer, Crit
server = Server(clients, model, n_workers)

server.run(n_rounds=n_rounds, eval_every=eval_every, n_rounds_no_aggregation=n_rounds_no_aggregation)

print('kas cas, cla, xzc')
dt = datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S')
os.makedirs(f'results/{dt}', exist_ok=True)
os.system(f'mv log results/{dt}/log')
Expand All @@ -46,7 +46,7 @@ def tune_hyperparameters():
def main():

# hyperparameters
n_rounds = 101
n_rounds = 301
eval_every = 10
n_rounds_no_aggregation = 201
Module = LogisticRegressor
Expand Down Expand Up @@ -90,6 +90,8 @@ def main():
# train
server.run(n_rounds=n_rounds, eval_every=eval_every, n_rounds_no_aggregation=n_rounds_no_aggregation)

print('run is done')

# # save results
# dt = datetime.datetime.now().strftime('%Y-%m-%d_%H:%M:%S')
# os.makedirs(f'results/{dt}', exist_ok=True)
Expand Down

0 comments on commit 273f791

Please sign in to comment.