Spaces:
Paused
Paused
AdriBat1 commited on
Commit Β·
1b272e7
1
Parent(s): 224ebad
Reorganize client files into examples, tests, and output folders
Browse files- remote-gpu-client/{bench_gpu.py β examples/bench_gpu.py} +0 -0
- remote-gpu-client/{inference_local.py β examples/inference_local.py} +8 -1
- remote-gpu-client/{julia.py β examples/julia.py} +0 -0
- remote-gpu-client/{mandelbrot.py β examples/mandelbrot.py} +0 -0
- remote-gpu-client/{train_iris.py β examples/train_iris.py} +0 -0
- remote-gpu-client/{train_nn.py β examples/train_nn.py} +0 -0
- remote-gpu-client/{launcher.py β legacy/launcher.py} +0 -0
- remote-gpu-client/{simple_nn.json β output/simple_nn.json} +0 -0
- remote-gpu-client/{simple_nn_weights.json β output/simple_nn_weights.json} +0 -0
- remote-gpu-client/{test_cpu.py β tests/test_cpu.py} +0 -0
- remote-gpu-client/{test_gpu.py β tests/test_gpu.py} +0 -0
- remote-gpu-client/{test_lib.py β tests/test_lib.py} +0 -0
remote-gpu-client/{bench_gpu.py β examples/bench_gpu.py}
RENAMED
|
File without changes
|
remote-gpu-client/{inference_local.py β examples/inference_local.py}
RENAMED
|
@@ -44,10 +44,17 @@ def predict(model_weights, X):
|
|
| 44 |
|
| 45 |
def main():
|
| 46 |
model_path = "simple_nn_weights.json"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
print(f"π Loading weights from {model_path}...")
|
| 48 |
|
| 49 |
if not os.path.exists(model_path):
|
| 50 |
-
print("β Model weights not found! Run 'python train_nn.py' first.")
|
| 51 |
sys.exit(1)
|
| 52 |
|
| 53 |
try:
|
|
|
|
| 44 |
|
| 45 |
def main():
|
| 46 |
model_path = "simple_nn_weights.json"
|
| 47 |
+
|
| 48 |
+
# Check current dir or ../output/
|
| 49 |
+
if not os.path.exists(model_path):
|
| 50 |
+
alt_path = os.path.join("..", "output", "simple_nn_weights.json")
|
| 51 |
+
if os.path.exists(alt_path):
|
| 52 |
+
model_path = alt_path
|
| 53 |
+
|
| 54 |
print(f"π Loading weights from {model_path}...")
|
| 55 |
|
| 56 |
if not os.path.exists(model_path):
|
| 57 |
+
print("β Model weights not found! Run 'python examples/train_nn.py' first.")
|
| 58 |
sys.exit(1)
|
| 59 |
|
| 60 |
try:
|
remote-gpu-client/{julia.py β examples/julia.py}
RENAMED
|
File without changes
|
remote-gpu-client/{mandelbrot.py β examples/mandelbrot.py}
RENAMED
|
File without changes
|
remote-gpu-client/{train_iris.py β examples/train_iris.py}
RENAMED
|
File without changes
|
remote-gpu-client/{train_nn.py β examples/train_nn.py}
RENAMED
|
File without changes
|
remote-gpu-client/{launcher.py β legacy/launcher.py}
RENAMED
|
File without changes
|
remote-gpu-client/{simple_nn.json β output/simple_nn.json}
RENAMED
|
File without changes
|
remote-gpu-client/{simple_nn_weights.json β output/simple_nn_weights.json}
RENAMED
|
File without changes
|
remote-gpu-client/{test_cpu.py β tests/test_cpu.py}
RENAMED
|
File without changes
|
remote-gpu-client/{test_gpu.py β tests/test_gpu.py}
RENAMED
|
File without changes
|
remote-gpu-client/{test_lib.py β tests/test_lib.py}
RENAMED
|
File without changes
|