.

main.py

@@ -22,9 +22,9 @@ torch.backends.cudnn.allow_tf32 = True
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
-DEBUG_MODE = True
-DEBUG_MODE_2 = False
-NUM_MAX_FRAMES = 32
+DEBUG_MODE = False
+DEBUG_MODE_2 = True
+NUM_MAX_FRAMES = 2
 
 SCREEN_WIDTH = 512
 SCREEN_HEIGHT = 384