first commit

BasaltTrainer/.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

BasaltTrainer/.python-version

@@ -0,0 +1 @@
+3.11

BasaltTrainer/main.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from basalt-trainer!")
+
+
+if __name__ == "__main__":
+    main()

BasaltTrainer/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "basalt-trainer"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "datasets>=4.3.0",
+    "matplotlib>=3.10.7",
+    "numpy>=2.3.4",
+  "torch>=2.9.0",
+  "torchvision>=0.24.0",
+]
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+torchvision = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cu126"
+url = "https://download.pytorch.org/whl/cu126"
+explicit = true
+