code-loader 0.2.89.dev5__py3-none-any.whl → 0.2.90.dev30__py3-none-any.whl

code_loader/metric_calculator_parallelized.py

@@ -31,8 +31,7 @@ class MetricCalculatorParallelized(LeapLoaderParallelizedBase):
         while True:
             metric_id, metric_name, input_arg_name_to_tensor = metrics_to_process.get(block=True)
             try:
-                with tf.device('/cpu:0'):
-                    metric_result = leap_loader.metric_by_name()[metric_name].function(**input_arg_name_to_tensor)
+                metric_result = leap_loader.metric_by_name()[metric_name].function(**input_arg_name_to_tensor)
             except Exception as e:
                 leap_script_trace = traceback.format_exc().split('File "<string>"')[-1]
                 ready_samples.put(MetricSerializableError(metric_id, metric_name, leap_script_trace, str(e)))

code_loader/visualizer_calculator_parallelized.py

@@ -40,9 +40,8 @@ class VisualizerCalculatorParallelized(LeapLoaderParallelizedBase):
             index_in_batch, visualizer_id, visualizer_name, input_arg_name_to_tensor = \
                 visualizers_to_process.get(block=True)
             try:
-                with tf.device('/cpu:0'):
-                    visualizer_result = \
-                        leap_loader.visualizer_by_name()[visualizer_name].function(**input_arg_name_to_tensor)
+                visualizer_result = \
+                    leap_loader.visualizer_by_name()[visualizer_name].function(**input_arg_name_to_tensor)
             except Exception as e:
                 ready_visualizations.put(VisualizerSerializableError(
                     visualizer_id, visualizer_name, index_in_batch, str(e)))

{code_loader-0.2.89.dev5.dist-info → code_loader-0.2.90.dev30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 0.2.89.dev5
+Version: 0.2.90.dev30
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-0.2.89.dev5.dist-info → code_loader-0.2.90.dev30.dist-info}/RECORD

@@ -22,15 +22,15 @@ code_loader/leap_binder/__init__.py,sha256=Oe7DNj6RmgX-tMruOrI7GYvf3SyAnzOxugF2e
 code_loader/leap_binder/leapbinder.py,sha256=-Ny2YewT1jg36Q5AXK8F2Fmv34zxhbcXZp807jIcHws,6293
 code_loader/leap_loader_parallelized_base.py,sha256=NqJL9rmNNSEOMHpecEzsHdkN0EGn4ASRqNkSyQ3v56Q,4758
 code_loader/leaploader.py,sha256=wb-eiGw6p-CML8w3noESEjJLjXJ_AiHxfAlGxSBtjYA,18523
-code_loader/metric_calculator_parallelized.py,sha256=X1FHFlfPorjFVNbms19s_yItkKFja0b97wUJHNvvlxs,2233
+code_loader/metric_calculator_parallelized.py,sha256=yRnKM-IhlJ7ygLw3Cj10UuGnk1_av_4KpKyfPDT2ZMY,2187
 code_loader/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/metrics/default_metrics.py,sha256=xtvTskR7LC5pyKSlFcvlyBpFY9TSXNGVAi5taEPSI60,6727
 code_loader/samples_generator_parallelized.py,sha256=_t0lmf11y1Tas4bdndOSk2b2QMjBQt9N_voTg-tcTuQ,2376
 code_loader/utils.py,sha256=WUcM97OuxrhfLCRPoH9EbXrxajNpYgX1CTMc3_PXtYU,1736
-code_loader/visualizer_calculator_parallelized.py,sha256=dKAixQ7-TIcG2sEkvbDbxKyuZP0WJP5oLO2bqN2mWRY,2597
+code_loader/visualizer_calculator_parallelized.py,sha256=WHXHwJfFuLI0sSvynULZTOkn_zHVTv1KjEw_GeYai1k,2547
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-0.2.89.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-0.2.89.dev5.dist-info/METADATA,sha256=BLUhtCyG9GOKIGAtJpYFIQn1-WG0tEwMQ2ElREE5n4s,955
-code_loader-0.2.89.dev5.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-code_loader-0.2.89.dev5.dist-info/RECORD,,
+code_loader-0.2.90.dev30.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-0.2.90.dev30.dist-info/METADATA,sha256=ky3mhJXhE3U0l_CfIUnydwrjjk3X4z4scNKPOqRrPT4,956
+code_loader-0.2.90.dev30.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+code_loader-0.2.90.dev30.dist-info/RECORD,,