Address PEP517 warning, minor requirements and documentation updates (a…

…nalogdevicesinc#328) * Minor documentation and requirements update * Address PEP517 warning * Add to known issues
oguzhanbsolak · Aug 28, 2024 · c6904ef · c6904ef
1 parent 6a4de6e
commit c6904ef
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 8 deletions.
diff --git a/KNOWN_ISSUES.txt b/KNOWN_ISSUES.txt
@@ -2,4 +2,5 @@ KNOWN ISSUES for ai8x-training with PyToch 2.3.1
 ================================================
 
 * As of PyTorch 2.3.1, torch.compile() requires CUDA.
-
+* When running on Windows, train.py may require the switch "--compiler-mode none" if there is "no working triton installation".
+  See also https://github.com/pytorch/pytorch/issues/122094
diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 # ADI MAX78000/MAX78002 Model Training and Synthesis
 
-July 5, 2024
+July 22, 2024
 
 **Note: This branch requires PyTorch 2. Please see the archive-1.8 branch for PyTorch 1.8 support. [KNOWN_ISSUES](KNOWN_ISSUES.txt) contains a list of known issues.**
 
@@ -421,7 +421,7 @@ For minor updates, pull the latest code and install the updated wheels:
 (ai8x-training) $ git pull
 (ai8x-training) $ git submodule update --init
 (ai8x-training) $ pip3 install -U pip setuptools
-(ai8x-training) $ pip3 install -U -r requirements.txt  # or requirements-xxx.txt, as shown above
+(ai8x-training) $ pip3 install -U -r requirements.txt  # add --extra-index-url if needed, as shown above
 ```
 
 ##### MSDK Updates

diff --git a/README.pdf b/README.pdf
diff --git a/distiller b/distiller
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,5 +1,5 @@
 [build-system]
-requires = ["pip", "setuptools", "wheel"]
+requires = ["pip", "setuptools >= 64", "wheel"]
 
 [project]
 name = "ai8x-training"

diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ torchaudio==2.3.1
 GitPython==3.1.43
 Pillow==10.4.0
 PyYAML==6.0.1
-albumentations==1.4.10
+albumentations==1.4.12
 faiss-cpu==1.8.0.post1
 batch-face==1.4.0
 h5py==3.11.0
@@ -13,7 +13,7 @@ numpy==1.26.4
 opencv-python==4.10.0.84
 pycocotools==2.0.8
 pyffmpeg==2.4.2.18.1
-pytorch-metric-learning==2.5.0
+pytorch-metric-learning==2.6.0
 pytube==15.0.0
 qrcode==7.4.2
 scipy==1.14.0
@@ -22,4 +22,4 @@ tensorboard==2.17.0
 tk==0.1.0
 torchmetrics==1.4.0.post0
 torchnet==0.0.4
--e distiller
+-e distiller --config-settings editable_mode=strict