From 9659a543588898594b01d516698d1587d0a1eae9 Mon Sep 17 00:00:00 2001 From: Arthur Douillard Date: Thu, 18 Jul 2024 22:31:13 +0200 Subject: [PATCH] Fix code example in README --- README.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/README.md b/README.md index 9a5fd93..a05bed4 100644 --- a/README.md +++ b/README.md @@ -255,10 +255,9 @@ Specifically, if you want to use OpenDiLoCo in your own training script, you can from functools import partial from open_diloco.hivemind_diloco import DiLoCoOptimizer - -import from hivemind.dht.dht import DHT + dht = DHT(start=True, initial_peers=os.environ["PEERS"]) inner_optimizer = partial(torch.optim.AdamW, lr=4e-4) # optimizer need to be function