imt-ring 1.3.10__py3-none-any.whl → 1.3.12__py3-none-any.whl

{imt_ring-1.3.10.dist-info → imt_ring-1.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.3.10
+Version: 1.3.12
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring

{imt_ring-1.3.10.dist-info → imt_ring-1.3.12.dist-info}/RECORD

@@ -15,7 +15,7 @@ ring/algorithms/custom_joints/rr_joint.py,sha256=jnRtjtOCALMaq2_0bcu2d7qgfQ6etXp
 ring/algorithms/custom_joints/suntay.py,sha256=7-kym1kMDwqYD_2um1roGcBeB8BlTCPe1wljuNGNARA,16676
 ring/algorithms/generator/__init__.py,sha256=p4ucl0zQtp5NwNoXIRjmTzGGRu2WOAWFfNmYRPwQles,912
 ring/algorithms/generator/base.py,sha256=sr-YZkjd8pZJAI5vFG_IqOO4AEeiEYtXr8uUsPMS6Q4,14779
-ring/algorithms/generator/batch.py,sha256=bslFSN2Gs_aX9cNwFooExhKUwevc70q3bspEMTwygm4,9256
+ring/algorithms/generator/batch.py,sha256=glsyedOAaHBMhFBLtMGJ9u1jHsJ2GITm0I5T0z2MFqU,9392
 ring/algorithms/generator/motion_artifacts.py,sha256=_kiAl1VHoX1fW5AUlXOtPBWyHIIFof_M78AP-m9f1ME,8790
 ring/algorithms/generator/pd_control.py,sha256=XJ_Gd5AkIRh-jBrMfQyMXjVwhx2gCNHznjzFbmAwhZs,5767
 ring/algorithms/generator/randomize.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3699
@@ -73,13 +73,13 @@ ring/sys_composer/inject_sys.py,sha256=Mj-q-mUjXKwkg-ol6IQAjf9IJfk7pGhez0_WoTKTg
 ring/sys_composer/morph_sys.py,sha256=2GpPtS5hT0eZMptdGpt30Hc97OykJNE67lEVRf7sHrc,12700
 ring/utils/__init__.py,sha256=FZ9ziQrWlx16QIpQ8RdLKrvN_17CAdvnZMNNodxWY0o,812
 ring/utils/backend.py,sha256=cKSi9sB59texqKzNVASTDczGKLCBL8VVDiP7TNdj41k,1294
-ring/utils/batchsize.py,sha256=3sLrmQMvCyoJ0Qi1uVzApSMfleYrIFAG7Y5_KuhJ-k4,1554
+ring/utils/batchsize.py,sha256=ByXGX7bw2gwrVirEsazm2JXnwNPGnqgEirzziYoSUS0,1553
 ring/utils/colab.py,sha256=ZLHwP0jNQUsmZJU4l68a5djULPi6T-jYNNHevjIoMn8,1631
 ring/utils/hdf5.py,sha256=BzXwVypZmEZeHVgeGZ78YYdi10NEQtnPhdrb8dQAXo0,5856
 ring/utils/normalizer.py,sha256=67L2BU1MRsMT4pD41ta3JJMppLN0ozFmnwrmXDtnqrQ,1698
 ring/utils/path.py,sha256=hAfSlqRi-ew536RnjDDM7IKapdMJc-EvhrR0Y-BCFWc,1265
 ring/utils/utils.py,sha256=mIcKNv5v2de8HrG7bAhl2bNfmwkMZyIIwFkJq2XWMOI,5357
-imt_ring-1.3.10.dist-info/METADATA,sha256=4z1vTo1XeXzJVPcAHPa9Y8psVMFF4BB2lpm9G82aADY,3105
-imt_ring-1.3.10.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-imt_ring-1.3.10.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.3.10.dist-info/RECORD,,
+imt_ring-1.3.12.dist-info/METADATA,sha256=VLLTMKRsKFxasZx_HTAIbKzrOQfjxMm7BW1zDuHG6oI,3105
+imt_ring-1.3.12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+imt_ring-1.3.12.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.3.12.dist-info/RECORD,,

ring/algorithms/generator/batch.py

@@ -98,7 +98,12 @@ def batch_generators_eager_to_list(
         jit = True if n_calls > 1 else False
         gen_jit = batch_generators_lazy(gen, size, jit=jit)
 
-        for _ in range(n_calls):
+        for _ in tqdm(
+            range(n_calls),
+            desc="number of calls for each generator",
+            total=n_calls,
+            leave=False,
+        ):
             key, consume = jax.random.split(key)
             sample = gen_jit(consume)
             # converts also to numpy; but with np.array.flags.writeable = False

ring/utils/batchsize.py

@@ -12,7 +12,7 @@ def batchsize_thresholds():
         eager_threshold = 4
     elif backend == "gpu":
         vmap_size_min = 8
-        eager_threshold = 128
+        eager_threshold = 32
     else:
         raise Exception(
             f"Backend {backend} has no default values, please add them in this function"