avoid looping over entire ds_train

mit-ll · May 14, 2024 · e3505af · e3505af
1 parent 9d113d1
commit e3505af
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/tornado_detection/train_tornado_tf.py b/scripts/tornado_detection/train_tornado_tf.py
@@ -122,6 +122,7 @@ def main(config):
     for x,y,w in ds_train:
         shp=get_shape(x)
         c_shp=x['coordinates'].shape
+        break
     in_shapes = (None,None,shp[-1])
     c_shapes = (None,None,c_shp[-1])