Update train.py

train.py

@@ -21,7 +21,7 @@ def main():
                 exploration_final_eps=0.02)
 
     # Define the number of training timesteps
-    TIMESTEPS = 100000  # Adjust as needed
+    TIMESTEPS = 550000  # Adjust as needed
 
     # Initialize the callback
     callback = SaveFramesCallback(save_freq=5000, save_path="models/frames", verbose=1)