diff --git a/pyproject.toml b/pyproject.toml index 52350049d..0fdd301fe 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,10 +14,10 @@ keywords = ["machine-learning", "ai", "deep-learning", "artificial-intelligence" dynamic = ["version"] [project.urls] -Homepage = "https://github.com/MaximIntegratedAI" -Documentation = "https://github.com/MaximIntegratedAI/ai8x-training/blob/develop/README.md" -Repository = "https://github.com/MaximIntegratedAI/ai8x-training.git" -Issues = "https://github.com/MaximIntegratedAI/ai8x-training/issues" +Homepage = "https://github.com/analogdevicesinc/MaximAI_Documentation" +Documentation = "https://github.com/analogdevicesinc/ai8x-training/blob/develop/README.md" +Repository = "https://github.com/analogdevicesinc/ai8x-training.git" +Issues = "https://github.com/analogdevicesinc/ai8x-training/issues" [project.scripts] train = "train:main" diff --git a/requirements.txt b/requirements.txt index 477a315dc..63a1792cd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,13 +5,12 @@ torchaudio==2.3.0 GitPython==3.1.43 Pillow==10.3.0 PyYAML==6.0.1 -albumentations==1.4.6 +albumentations==1.4.7 faiss-cpu==1.8.0 face-detection==0.2.2 h5py==3.11.0 -imutils==0.5.4 kornia==0.7.2 -librosa==0.10.2 +librosa==0.10.2.post1 numpy==1.26.4 opencv-python==4.9.0.80 pycocotools==2.0.7 @@ -21,6 +20,6 @@ pytube==15.0.0 scipy==1.13.0 tensorboard==2.16.2 tk==0.1.0 -torchmetrics==1.4.0 +torchmetrics==1.4.0.post0 torchnet==0.0.4 -e distiller