minor fixes

.gitignore

@@ -1,6 +1,6 @@
 # system
 *.egg-info/
-__pycache__/
+*__pycache__/
 *.py[cod]
 *.eggs
 build
@@ -19,4 +19,5 @@ esempi
 examples
 src/models/yolov5
 src/models/checkpoints
-prova*
+prova*
+notebooks

README.md

@@ -13,10 +13,11 @@ The model and pipeline aims at being task-agnostic on the domain of Document Und
 - [ ] Transform Doc2Graph into a PyPI package
 
 Roadmap:
-- [Environment Setup](#environment-setup)
-- [Training](#training)
-- [Testing](#testing)
-- [Cite this project](#cite-this-project)
+- [`Doc2Graph`](#doc2graph)
+  - [Environment Setup](#environment-setup)
+  - [Training](#training)
+  - [Testing](#testing)
+  - [Cite this project](#cite-this-project)
 
 ## Environment Setup
 Setup the initial conda environment
@@ -27,12 +28,12 @@ conda activate doc2graph
 cd doc2graph
 ```
 
-Then, install [setuptools-git-versioning](https://pypi.org/project/setuptools-git-versioning/) and doc2graph package itself.
+Then, install [setuptools-git-versioning](https://pypi.org/project/setuptools-git-versioning/) and doc2graph package itself. The following has been tested only on linux: for different OS installations refer directly to [PyTorch](https://pytorch.org/get-started/previous-versions/) and [DGL](https://www.dgl.ai/pages/start.html) original documentation.
 
 ```
+pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
+pip install dgl-cu113 dglgo -f https://data.dgl.ai/wheels/repo.html
 pip install setuptools-git-versioning && pip install -e .
-pip install dglgo==0.0.1
-pip install dgl-cu113==0.8.2 -f https://data.dgl.ai/wheels/repo.html
 python -m spacy download en_core_web_lg
 ```