Small updates to website (facebookresearch#157)

Summary: small updates in tutorial and website getting started Pull Request resolved: facebookresearch#157 Reviewed By: bottler Differential Revision: D26125920 Pulled By: prigoyal fbshipit-source-id: 724d7e122e90c877289389bf90ccad4d5d925e14
qianrenjian · Jan 28, 2021 · cbb06ff · cbb06ff
1 parent ab44d01
commit cbb06ff
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/tutorials/Installation.ipynb b/tutorials/Installation.ipynb
@@ -82,7 +82,7 @@
         "conda create -n vissl python=3.8\n",
         "conda activate vissl\n",
         "conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.2\n",
-        "conda install -c vissl apex vissl\n",
+        "conda install -c vissl -c iopath -c conda-forge -c pytorch -c defaults apex vissl\n",
         "```\n",
         "\n",
         "For other versions of PyTorch, Python, CUDA, please modify the above instructions with the\n",
@@ -175,7 +175,7 @@
         "\n",
         "**Step 4: Install VISSL**\n",
         "\n",
-        "```\n",
+        "```bash\n",
         "# clone vissl repository\n",
         "cd $HOME && git clone --recursive https://github.com/facebookresearch/vissl.git && cd $HOME/vissl/\n",
         "# install vissl dependencies\n",

diff --git a/website/pages/en/index.js b/website/pages/en/index.js
@@ -111,14 +111,21 @@ class Index extends React.Component {
     const pre = '```';
 
     const codeExample = `${pre}bash
-python tools/run_distributed_engines.py config=test/integration_test/quick_simclr
+python3 run_distributed_engines.py config=quick_1gpu_resnet50_simclr config.DATA.TRAIN.DATA_SOURCES=[synthetic]
+    `;
+
+    const downloadBlock = `${pre}bash
+cd /tmp/ && mkdir -p /tmp/configs/config
+wget -q -O configs/__init__.py https://dl.fbaipublicfiles.com/vissl/tutorials/configs/__init__.py
+wget -q -O configs/config/quick_1gpu_resnet50_simclr.yaml https://dl.fbaipublicfiles.com/vissl/tutorials/configs/quick_1gpu_resnet50_simclr.yaml
+wget -q  https://dl.fbaipublicfiles.com/vissl/tutorials/run_distributed_engines.py
     `;
 
     const installBlock = `${pre}bash
 conda create -n vissl python=3.8
 conda activate vissl
 conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.2
-conda install -c vissl apex vissl
+conda install -c vissl -c iopath -c conda-forge -c pytorch -c defaults apex vissl
     `;
 
     const QuickStart = () => (
@@ -132,11 +139,14 @@ conda install -c vissl apex vissl
             <li>
               <h4>Install VISSL:</h4>
               <a>via conda:</a>
-              {/* <MarkdownBlock>{bash`conda install vissl -c vissl`}</MarkdownBlock> */}
               <MarkdownBlock>{installBlock}</MarkdownBlock>
             </li>
             <li>
-              <h4>Try training SimCLR model  </h4>
+              <h4>Download SimCLR yaml config and builtin distributed launcher:  </h4>
+              <MarkdownBlock>{downloadBlock}</MarkdownBlock>
+            </li>
+            <li>
+              <h4>Try training SimCLR model on 1-gpu:  </h4>
               <MarkdownBlock>{codeExample}</MarkdownBlock>
             </li>
           </ol>