new lines

official/vision/beta/data/tfrecord_lib.py

@@ -139,7 +139,7 @@ def write_tf_record_dataset(output_path, annotation_iterator,
   total_num_annotations_skipped = 0
 
   if use_multiprocessing:
-    pool = mp.Pool()
+    pool = mp.Pool(z)
     if unpack_arguments:
       tf_example_iterator = pool.starmap(process_func, annotation_iterator)
     else:

official/vision/beta/projects/yolo/modeling/backbones/darknet.py

@@ -682,4 +682,4 @@ def build_darknet(
       norm_epsilon=norm_activation_config.norm_epsilon,
       kernel_regularizer=l2_regularizer)
   model.summary()
-  return model
+  return model
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/backbones/darknet_test.py

@@ -126,4 +126,4 @@ class DarknetTest(parameterized.TestCase, tf.test.TestCase):
     self.assertAllEqual(network.get_config(), new_network.get_config())
 
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py

@@ -475,4 +475,4 @@ class YoloDecoder(tf.keras.Model):
 
   @classmethod
   def from_config(cls, config, custom_objects=None):
-    return cls(**config)
+    return cls(**config)
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder_test.py

@@ -150,4 +150,4 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
     self.assertAllEqual(decoder.get_config(), decoder_from_config.get_config())
 
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/heads/yolo_head.py

@@ -119,4 +119,4 @@ class YoloHead(tf.keras.layers.Layer):
 
   @classmethod
   def from_config(cls, config, custom_objects=None):
-    return cls(**config)
+    return cls(**config)
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/heads/yolo_head_test.py

@@ -70,4 +70,4 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
     self.assertAllEqual(head.get_config(), head_from_config.get_config())
 
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py

@@ -1640,4 +1640,4 @@ class DarkRouteProcess(tf.keras.layers.Layer):
     if self._csp_stack > 0:
       return self._call_csp(inputs, training=training)
     else:
-      return self._call_regular(inputs)
+      return self._call_regular(inputs)
\ No newline at end of file

official/vision/beta/projects/yolo/modeling/layers/nn_blocks_test.py

@@ -384,4 +384,4 @@ class DarkRouteProcessTest(tf.test.TestCase, parameterized.TestCase):
     return
 
 if __name__ == '__main__':
-  tf.test.main()
+  tf.test.main()
\ No newline at end of file