setup train

check_space.py

@@ -0,0 +1,37 @@
+# from huggingface_hub import HfApi
+
+# my_token = """  
+# api = HfApi(token=my_token)
+
+# spaces = api.list_spaces(author="uav-vts")  # hoặc namespace bạn được cấp
+# for space in spaces:
+#     print(space.id, getattr(space, "private", "N/A"), getattr(space, "sdk", "N/A"))
+
+
+from huggingface_hub import HfApi
+
+my_token = ""  
+
+my_namespace = "uav-vts"  
+
+api = HfApi(token=my_token)
+
+spaces = api.list_spaces(author=my_namespace)
+
+if not spaces:
+    print("Không tìm thấy Space nào trong namespace / resource group này.")
+else:
+    print(f"Tìm thấy {len(list(spaces))} Space(s) trong namespace '{my_namespace}':\n")
+    for space in spaces:
+        owner, name = space.id.split("/") if "/" in space.id else ("unknown", space.id)
+        url = f"https://huggingface.co/spaces/{space.id}"
+        private = getattr(space, "private", "N/A")
+        sdk = getattr(space, "sdk", "N/A")
+        last_modified = getattr(space, "lastModified", "N/A")
+        # Trạng thái có thể lấy từ cardData nếu expand=["cardData"]
+        status = getattr(getattr(space, "cardData", {}), "status", "N/A")
+        
+        print(f"Owner: {owner}, Name: {name}")
+        print(f"URL: {url}")
+        print(f"Private: {private}, SDK: {sdk}, LastModified: {last_modified}, Status: {status}")
+        print("-" * 60)

train_hugg.py

@@ -0,0 +1,32 @@
+# from huggingface_hub import HfApi
+
+# api = HfApi()
+
+# job = api.create_job(
+#     repo_id="uav-vts/training_sem",
+#     docker_image="pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel",
+#     command="python runner/train.py",
+#     hardware="a100",   
+#     # token="<TOKEN>"
+# )
+
+# print("Job created:", job.job_id)
+
+
+from huggingface_hub import run_job
+
+image = "pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel"
+command = [
+    "python",
+    "-c",
+    "import torch; print(f'This code ran with the following GPU: {torch.cuda.get_device_name()}')"
+]
+
+job = run_job(
+    image=image,
+    command=command,
+    flavor="a100-large",   
+    namespace="uav-vts"
+)
+print("Job submitted!")
+print("Track it here:", job.url)

train_job.py

@@ -0,0 +1,19 @@
+from huggingface_hub import run_job
+
+
+# docker pull pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
+job = run_job(
+    image="pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel",
+    command=[
+            "bash",
+            "-c",
+            "apt-get update && apt-get install -y git && "
+            "git clone https://huggingface.co/uav-vts/training_sem . && "
+            "pwd && ls -al && "
+            "pip install -r requirements.txt && "
+            "python runner/train.py"
+    ],
+    flavor="a100-large",
+    namespace="uav-vts"
+    # token="hf_xxx"
+)