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} 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