initial commit

2026-05-29 18:49:01 +08:00
commit 9f0321eff8
21 changed files with 3143 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,17 @@
+# ===== Core =====
+numpy
+scipy
+
+# ===== PyTorch (CUDA 12.4) =====
+# Install via: pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
+torch>=2.4
+torchvision
+
+# ===== Data loading =====
+webdataset
+opencv-python
+
+# ===== Training / logging =====
+tensorboard
+tqdm
+matplotlib