Bladeren bron

Move CLI server launch script to hivemind/hivemind_cli (#173)

Max Ryabinin 4 jaren geleden
bovenliggende
commit
bbcbd786e5
5 gewijzigde bestanden met toevoegingen van 5 en 5 verwijderingen
  1. 3 3
      docs/user/quickstart.md
  2. 1 1
      hivemind/__init__.py
  3. 0 0
      hivemind/hivemind_cli/config.yml
  4. 0 0
      hivemind/hivemind_cli/run_server.py
  5. 1 1
      setup.py

+ 3 - 3
docs/user/quickstart.md

@@ -26,11 +26,11 @@ You can also install it in the editable mode with `pip install -e .`.
 
 `hivemind.Server` hosts one or several experts (PyTorch modules) for remote access. These experts are responsible for
 most of the model parameters and computation. The server can be started using either Python or
-[a shell script](https://github.com/learning-at-home/hivemind/blob/master/scripts/run_server.py). We'll use the shell
+[a shell script](https://github.com/learning-at-home/hivemind/blob/master/hivemind/hivemind_cli/run_server.py). We'll use the shell
 for now. To host a server with default experts, run this in your shell:
 
 ```sh
-python scripts/run_server.py --expert_cls ffn --hidden_dim 512 --num_experts 5 --expert_pattern "expert.[0:5]" \
+python hivemind/hivemind_cli/run_server.py --expert_cls ffn --hidden_dim 512 --num_experts 5 --expert_pattern "expert.[0:5]" \
                              --listen_on 0.0.0.0:1337 --dht_port 1338
 # note: if you omit listen_on and/or dht_port, they will be chosen automatically and printed to stdout.
 ```
@@ -79,7 +79,7 @@ architecture [here](https://github.com/learning-at-home/hivemind/blob/master/hiv
 You can create additional servers in the same decentralized network using `--initial_peers` argument:
 
 ```sh
-python scripts/run_server.py --expert_cls ffn --hidden_dim 512 --num_experts 10 --expert_pattern "expert.[5:250]" \
+python hivemind/hivemind_cli/run_server.py --expert_cls ffn --hidden_dim 512 --num_experts 10 --expert_pattern "expert.[5:250]" \
                               --initial_peers localhost:1338
 ```
 

+ 1 - 1
hivemind/__init__.py

@@ -3,4 +3,4 @@ from hivemind.dht import *
 from hivemind.server import *
 from hivemind.utils import *
 
-__version__ = '0.9.4'
+__version__ = '0.9.5'

+ 0 - 0
scripts/config.yml → hivemind/hivemind_cli/config.yml


+ 0 - 0
scripts/run_server.py → hivemind/hivemind_cli/run_server.py


+ 1 - 1
setup.py

@@ -94,7 +94,7 @@ setup(
         'Topic :: Software Development :: Libraries :: Python Modules',
     ],
     entry_points={
-        'console_scripts': ['hivemind-server = scripts.run_server:main', ]
+        'console_scripts': ['hivemind-server = hivemind.hivemind_cli.run_server:main', ]
     },
     # What does your project relate to?
     keywords='pytorch, deep learning, machine learning, gpu, distributed computing, volunteer computing, dht',