Antigravity Agent commited on
Commit
1ec970c
·
1 Parent(s): f7709b7

Deploy Fix: Disable interactive build

Browse files
Files changed (1) hide show
  1. train_hf.py +2 -2
train_hf.py CHANGED
@@ -22,7 +22,7 @@ def train_hf(repo_id, token, total_timesteps=500000):
22
  # Checkpoint Callback
23
  checkpoint_callback = CheckpointCallback(
24
  save_freq=50000,
25
- save_path='./checkpoints/',
26
  name_prefix='liquid_ppo_drone'
27
  )
28
 
@@ -30,7 +30,7 @@ def train_hf(repo_id, token, total_timesteps=500000):
30
  model.learn(total_timesteps=total_timesteps, callback=checkpoint_callback)
31
 
32
  # Save Final Model
33
- model_path = "liquid_ppo_drone_final.zip"
34
  model.save(model_path)
35
  print(f"Model saved to {model_path}")
36
 
 
22
  # Checkpoint Callback
23
  checkpoint_callback = CheckpointCallback(
24
  save_freq=50000,
25
+ save_path='/tmp/checkpoints/',
26
  name_prefix='liquid_ppo_drone'
27
  )
28
 
 
30
  model.learn(total_timesteps=total_timesteps, callback=checkpoint_callback)
31
 
32
  # Save Final Model
33
+ model_path = "/tmp/liquid_ppo_drone_final.zip"
34
  model.save(model_path)
35
  print(f"Model saved to {model_path}")
36