reflectorch 1.2.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

reflectorch/models/networks/residual_net.py

@@ -18,15 +18,21 @@ class ResidualMLP(nn.Module):
             repeats_per_block: int = 2,
             activation: str = 'relu',
             use_batch_norm: bool = True,
+            use_layer_norm: bool = False,
             dropout_rate: float = 0.0,
             residual: bool = True,
             adaptive_activation: bool = False,
             conditioning: str = 'glu',
+            concat_condition_first_layer: bool = True,
+            film_with_tanh: bool = False,
     ):
         super().__init__()
 
-        dim_first_layer = dim_in + dim_condition
+        self.concat_condition_first_layer = concat_condition_first_layer
+
+        dim_first_layer = dim_in + dim_condition if concat_condition_first_layer else dim_in
         self.first_layer = nn.Linear(dim_first_layer, layer_width)
+
         self.blocks = nn.ModuleList(
             [
                 ResidualBlock(
@@ -35,24 +41,28 @@ class ResidualMLP(nn.Module):
                     repeats_per_block=repeats_per_block,
                     activation=activation,
                     use_batch_norm=use_batch_norm,
+                    use_layer_norm=use_layer_norm,
                     dropout_rate=dropout_rate,
                     residual=residual,
                     adaptive_activation=adaptive_activation,
                     conditioning = conditioning,
+                    film_with_tanh = film_with_tanh,
                 )
                 for _ in range(num_blocks)
             ]
         )
+
         self.last_layer = nn.Linear(layer_width, dim_out)
 
     def forward(self, x, condition=None):
-        if condition is None:
-            x = self.first_layer(x)
-        else:
+        if self.concat_condition_first_layer and condition is not None:
             x = self.first_layer(torch.cat([x, condition], dim=-1))
+        else:
+            x = self.first_layer(x)
 
         for block in self.blocks:
             x = block(x, condition=condition)
+
         x = self.last_layer(x)
 
         return x
@@ -68,19 +78,23 @@ class ResidualBlock(nn.Module):
             repeats_per_block: int = 2,
             activation: str = 'relu',
             use_batch_norm: bool = False,
+            use_layer_norm: bool = False,
             dropout_rate: float = 0.0,
             residual: bool = True,
             adaptive_activation: bool = False,
             conditioning: str = 'glu',
+            film_with_tanh: bool = False,
     ):
         super().__init__()
          
         self.residual = residual
         self.repeats_per_block = repeats_per_block
         self.use_batch_norm = use_batch_norm
+        self.use_layer_norm = use_layer_norm
         self.dropout_rate = dropout_rate
         self.adaptive_activation = adaptive_activation
         self.conditioning = conditioning
+        self.film_with_tanh = film_with_tanh
 
         if not adaptive_activation:
             self.activation = activation_by_name(activation)()
@@ -92,7 +106,11 @@ class ResidualBlock(nn.Module):
         if use_batch_norm:
             self.batch_norm_layers = nn.ModuleList(
                 [nn.BatchNorm1d(layer_width, eps=1e-3) for _ in range(repeats_per_block)]
-            )
+             )
+        elif use_layer_norm:
+            self.layer_norm_layers = nn.ModuleList(
+                [nn.LayerNorm(layer_width) for _ in range(repeats_per_block)]
+             )
 
         if dim_condition:
             if conditioning == 'glu':
@@ -113,12 +131,17 @@ class ResidualBlock(nn.Module):
         for i in range(self.repeats_per_block):
             if self.use_batch_norm:
                 x = self.batch_norm_layers[i](x)
+            elif self.use_layer_norm:
+                x = self.layer_norm_layers[i](x)
+
             if not self.adaptive_activation:
                 x = self.activation(x)
             else:
                 x = self.activation_layers[i](x)
+
             if self.dropout_rate > 0 and i == self.repeats_per_block - 1:
                 x = self.dropout(x)
+
             x = self.linear_layers[i](x)
         
         if condition is not None:
@@ -126,6 +149,9 @@ class ResidualBlock(nn.Module):
                 x = F.glu(torch.cat((x, self.condition_layer(condition)), dim=-1), dim=-1)
             elif self.conditioning == 'film':
                 gamma, beta = torch.chunk(self.condition_layer(condition), chunks=2, dim=-1)
+                if self.film_with_tanh:
+                    tanh = nn.Tanh()
+                    gamma, beta = tanh(gamma), tanh(beta)
                 x = x * gamma + beta
 
         return x0 + x if self.residual else x

reflectorch/runs/train.py

@@ -72,7 +72,6 @@ def run_train_on_cluster(config_name: str):
 
 def _change_to_test_config(config, batch_size: int, num_iterations: int):
     config = dict(config)
-    config['training']['logger']['use_neptune'] = False
     config['training']['num_iterations'] = num_iterations
     config['training']['batch_size'] = batch_size
     config['training']['update_tqdm_freq'] = 1

reflectorch/runs/utils.py

@@ -136,7 +136,7 @@ def get_callbacks_from_config(config: dict, folder_paths: dict = None) -> Tuple[
         if callback:
             callbacks.append(callback)
 
-    if train_conf['logger']['use_neptune']:
+    if 'logger' in train_conf.keys():
         callbacks.append(LogLosses())
 
     return tuple(callbacks)
@@ -162,9 +162,9 @@ def get_trainer_from_config(config: dict, folder_paths: dict = None):
 
     optim_cls = getattr(torch.optim, train_conf['optimizer'])
 
-    logger = None
+    logger_conf = train_conf.get('logger', None)
+    logger = init_from_conf(logger_conf) if logger_conf and logger_conf.get('cls') else None 
 
-    train_with_q_input = train_conf.get('train_with_q_input', False)
     clip_grad_norm_max = train_conf.get('clip_grad_norm_max', None)
 
     trainer_cls = globals().get(train_conf['trainer_cls']) if 'trainer_cls' in train_conf else PointEstimatorTrainer
@@ -173,10 +173,13 @@ def get_trainer_from_config(config: dict, folder_paths: dict = None):
 
     trainer = trainer_cls(
         model, dset, train_conf['lr'], train_conf['batch_size'], clip_grad_norm_max=clip_grad_norm_max,
-        logger=logger, optim_cls=optim_cls, train_with_q_input=train_with_q_input, 
+        logger=logger, optim_cls=optim_cls, 
         **trainer_kwargs
     )
 
+    if train_conf.get('train_with_q_input', False) and getattr(trainer, 'train_with_q_input', None) is not None: #only for back-compatibility with configs in older versions
+        trainer.train_with_q_input = True
+
     return trainer
 
 
@@ -187,7 +190,7 @@ def get_trainer_by_name(config_name, config_dir=None, model_path=None, load_weig
     Args:
         config_name (str): name of the configuration file
         config_dir (str): path of the configuration directory
-        model_path (str, optional): path to the network weights.
+        model_path (str, optional): path to the network weights. The default path is 'saved_models' located in the package directory
         load_weights (bool, optional): if True the saved network weights are loaded into the network. Defaults to True.
         inference_device (str, optional): overwrites the device in the configuration file for the purpose of inference on a different device then the training was performed on. Defaults to 'cuda'.
 
@@ -195,8 +198,7 @@ def get_trainer_by_name(config_name, config_dir=None, model_path=None, load_weig
         Trainer: the trainer object
     """
     config = load_config(config_name, config_dir)
-    config['model']['network']['pretrained_name'] = None
-    config['training']['logger']['use_neptune'] = False
+    #config['model']['network']['pretrained_name'] = None
 
     config['model']['network']['device'] = inference_device
     config['dset']['prior_sampler']['kwargs']['device'] = inference_device
@@ -219,7 +221,7 @@ def get_trainer_by_name(config_name, config_dir=None, model_path=None, load_weig
 
     if str(model_path).endswith('.pt'):
         try:
-            state_dict = torch.load(model_path, map_location=inference_device)
+            state_dict = torch.load(model_path, map_location=inference_device, weights_only=False)
         except Exception as err:
             raise RuntimeError(f'Could not load model from {model_path}') from err
 
@@ -238,7 +240,7 @@ def get_trainer_by_name(config_name, config_dir=None, model_path=None, load_weig
         raise RuntimeError('Weigths file with unknown extension')
 
     return trainer
-
+ 
 def get_callbacks_by_name(config_name, config_dir=None):
     """Initializes the trainer callbacks based on a configuration file
 
@@ -362,6 +364,38 @@ def convert_pt_to_safetensors(input_dir):
 
             safetensors.torch.save_file(tensors=model_state_dict, filename=safetensors_file_path)
 
+def convert_files_to_safetensors(files):
+    """
+    Converts specified .pt files to .safetensors format.
+
+    Args:
+        files (str or list of str): Path(s) to .pt files containing model state dictionaries.
+    """
+    if isinstance(files, str):
+        files = [files]
+
+    for pt_file_path in files:
+        if not pt_file_path.endswith('.pt'):
+            print(f"Skipping {pt_file_path}: not a .pt file.")
+            continue
+
+        if not os.path.exists(pt_file_path):
+            print(f"File {pt_file_path} does not exist.")
+            continue
+
+        safetensors_file_path = pt_file_path[:-3] + '.safetensors'
+
+        if os.path.exists(safetensors_file_path):
+            print(f"Skipping {pt_file_path}: .safetensors version already exists.")
+            continue
+
+        print(f"Converting {pt_file_path} to .safetensors format.")
+        data_pt = torch.load(pt_file_path, weights_only=False)
+        model_state_dict = data_pt["model"]
+        model_state_dict = split_complex_tensors(model_state_dict)
+
+        safetensors.torch.save_file(tensors=model_state_dict, filename=safetensors_file_path)
+
 def load_state_dict_safetensors(model, filename, device):
     state_dict = safetensors.torch.load_file(filename=filename, device=device)
     state_dict = recombine_complex_tensors(state_dict)

{reflectorch-1.2.0.dist-info → reflectorch-1.3.0.dist-info}/METADATA

@@ -1,9 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: reflectorch
-Version: 1.2.0
+Version: 1.3.0
 Summary: A Pytorch-based package for the analysis of reflectometry data
 Author-email: Vladimir Starostin <vladimir.starostin@uni-tuebingen.de>, Valentin Munteanu <valentin.munteanu@uni-tuebingen.de>
 Maintainer-email: Valentin Munteanu <valentin.munteanu@uni-tuebingen.de>, Vladimir Starostin <vladimir.starostin@uni-tuebingen.de>, Alexander Hinderhofer <alexander.hinderhofer@uni-tuebingen.de>
+License-Expression: MIT
 Project-URL: Source, https://github.com/schreiber-lab/reflectorch/
 Project-URL: Issues, https://github.com/schreiber-lab/reflectorch/issues
 Project-URL: Documentation, https://schreiber-lab.github.io/reflectorch/
@@ -11,32 +12,33 @@ Keywords: reflectometry,machine learning
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Classifier: Environment :: GPU :: NVIDIA CUDA
-Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
 Classifier: Development Status :: 4 - Beta
 Classifier: Topic :: Scientific/Engineering :: Physics
 Classifier: Intended Audience :: Science/Research
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: numpy <2.0,>=1.18.1
-Requires-Dist: torch >=1.8.1
+Requires-Dist: numpy
+Requires-Dist: torch>=1.8.1
 Requires-Dist: scipy
 Requires-Dist: tqdm
 Requires-Dist: PyYAML
 Requires-Dist: click
 Requires-Dist: matplotlib
 Requires-Dist: ipywidgets
-Requires-Dist: huggingface-hub
+Requires-Dist: huggingface_hub
 Requires-Dist: safetensors
-Provides-Extra: build
-Requires-Dist: build ; extra == 'build'
-Requires-Dist: twine ; extra == 'build'
-Provides-Extra: docs
-Requires-Dist: jupyter-book ; extra == 'docs'
-Requires-Dist: sphinx ; extra == 'docs'
+Requires-Dist: tensorboard
 Provides-Extra: tests
-Requires-Dist: pytest ; extra == 'tests'
-Requires-Dist: pytest-cov ; extra == 'tests'
+Requires-Dist: pytest; extra == "tests"
+Requires-Dist: pytest-cov; extra == "tests"
+Provides-Extra: docs
+Requires-Dist: jupyter-book; extra == "docs"
+Requires-Dist: sphinx; extra == "docs"
+Provides-Extra: build
+Requires-Dist: build; extra == "build"
+Requires-Dist: twine; extra == "build"
+Dynamic: license-file
 
 # Reflectorch
 
@@ -50,7 +52,7 @@ Requires-Dist: pytest-cov ; extra == 'tests'
 [![Python version](https://img.shields.io/badge/python-3.7%7C3.8%7C3.9%7C3.10%7C3.11%7C3.12-blue.svg)](https://www.python.org/)
 ![CI workflow status](https://github.com/schreiber-lab/reflectorch/actions/workflows/ci.yml/badge.svg)
 ![Repos size](https://img.shields.io/github/repo-size/schreiber-lab/reflectorch)
-[![CodeFactor](https://www.codefactor.io/repository/github/schreiber-lab/reflectorch/badge)](https://www.codefactor.io/repository/github/schreiber-lab/reflectorch)
+<!-- [![CodeFactor](https://www.codefactor.io/repository/github/schreiber-lab/reflectorch/badge)](https://www.codefactor.io/repository/github/schreiber-lab/reflectorch) -->
 [![Jupyter Book Documentation](https://jupyterbook.org/badge.svg)](https://jupyterbook.org/)
 [![Documentation Page](https://img.shields.io/badge/Documentation%20Page-%23FFDD33.svg?style=flat&logo=read-the-docs&logoColor=black)](https://schreiber-lab.github.io/reflectorch/)
 <!-- [![Code style: Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff) -->
@@ -101,8 +103,8 @@ We provide an interactive Google Colab notebook for exploring the basic function
 [![Hugging Face](https://img.shields.io/badge/Hugging%20Face-%23FFD700.svg?style=flat&logo=huggingface&logoColor=black)](https://huggingface.co/valentinsingularity/reflectivity)
 Configuration files and the corresponding pretrained model weights are hosted on Huggingface: [https://huggingface.co/valentinsingularity/reflectivity](https://huggingface.co/valentinsingularity/reflectivity).
 
-[![Docker](https://img.shields.io/badge/Docker-2496ED.svg?style=flat&logo=docker&logoColor=white)](https://hub.docker.com/)
-Docker images for reflectorch *will* be hosted on Dockerhub.
+<!-- [![Docker](https://img.shields.io/badge/Docker-2496ED.svg?style=flat&logo=docker&logoColor=white)](https://hub.docker.com/)
+Docker images for reflectorch *will* be hosted on Dockerhub. -->
 
 
 ## Citation

{reflectorch-1.2.0.dist-info → reflectorch-1.3.0.dist-info}/RECORD

@@ -4,15 +4,15 @@ reflectorch/test_config.py,sha256=1T7pMJ-WYLEu-4WtYMQxcJrqXvgdpvJ1yi2qINd0kNA,99
 reflectorch/train.py,sha256=-c8ac1fpjrCiEwnAaXg_rcBNl1stO1V5p5afx_78xHs,87
 reflectorch/train_on_cluster.py,sha256=aG3_g5_rzL8iL1tvtdY9ueJTo1f2Pn8lGJgudrrRknU,109
 reflectorch/utils.py,sha256=ehG5dU02_WIFPIVGPsSjHl3Ji10MQ8-jsCwRkwUO7D4,1993
-reflectorch/data_generation/__init__.py,sha256=TcF7kf16GCsQpp6cIXj-s4vb_gjrtv7tgXwLSeI8Xy8,3353
-reflectorch/data_generation/dataset.py,sha256=wEHMIzA1XQGQkzkN8WQRNqG53lyzxIVc80nMJKs0mgI,7273
+reflectorch/data_generation/__init__.py,sha256=4FuuBfhzgEGh_YkFmLBqjyjhlqI3D5It8iu6DqGmf4g,3419
+reflectorch/data_generation/dataset.py,sha256=f2uYb8j8H88owZJhg9aMP7qen00_o2cXmDvA69yJwr8,8316
 reflectorch/data_generation/likelihoods.py,sha256=gnqmsEfsZnCC6WuPcIe9rFl4SeiLv_LyNlTVM-5YTE8,2862
-reflectorch/data_generation/noise.py,sha256=c4ytClId3t3T5FO8r0NqAVz-x_zYWAJ2VkpxcvssrVY,16159
+reflectorch/data_generation/noise.py,sha256=Ps9lDKUjtjcpUfJ_L9zTmlj88GyicjNs962lLVEg2aI,22090
 reflectorch/data_generation/process_data.py,sha256=iBvWB4_X5TTIZTacsq9baBZPyGCRyI1G-J3l1BVMuD4,1180
-reflectorch/data_generation/q_generator.py,sha256=Etlp_6Tj2it2nK8PQSWgQZBLh8_459P985GrZs9U1lE,8529
+reflectorch/data_generation/q_generator.py,sha256=_WGCqdu1M3kSUlSbAGMPXDIU82SBYT2pcmnUPqctkyo,9279
 reflectorch/data_generation/scale_curves.py,sha256=hNZoGkA9FMdLUarp0fwj5nCDYvpy9NbSiO85_rCaNr8,4084
-reflectorch/data_generation/smearing.py,sha256=MfHQ2sa-wPu5stO835m8i1-3NnJYvxS3qpzNs1jXWAE,3109
-reflectorch/data_generation/utils.py,sha256=3Di6I5Ihy2rm8NMeCEvIVqH0IXFEX2bcxkHSc6y9WgU,5607
+reflectorch/data_generation/smearing.py,sha256=ZQ4bzHegUOVhNWOTHAi3KfNau85bchp1FX1NV0Kp7s8,4538
+reflectorch/data_generation/utils.py,sha256=gYJFxSj_YYHXk8kCYIagGaugUv9SrxmZvGeR9gJQMVc,8500
 reflectorch/data_generation/priors/__init__.py,sha256=ZUaQUgNR44MQGYcPVJaSzzoE270710RqdKAPcaBNHWo,1941
 reflectorch/data_generation/priors/base.py,sha256=JNa2A3F4IWaEwV7gyTQjVdCRevUI43UUpTOm1U3XA8k,1705
 reflectorch/data_generation/priors/exp_subprior_sampler.py,sha256=hjHey_32HGgZ4ojOAhmmx8jcQPCyvcPRgxMlfthSdOo,11551
@@ -20,32 +20,35 @@ reflectorch/data_generation/priors/independent_priors.py,sha256=ZdFCW4Ea6cK9f0Pn
 reflectorch/data_generation/priors/multilayer_models.py,sha256=lAf-HJPbIDnbD1ecDTvx03TfA4jLN5tTLbwaBCiYgWM,7763
 reflectorch/data_generation/priors/multilayer_structures.py,sha256=b1LTkzMK_R2hgbAmif96zRRzZRV2V-dD7eSTaNalMU8,3693
 reflectorch/data_generation/priors/no_constraints.py,sha256=qioTThJ17NixCaMIIf4dusPtvcK_uI2jOoysyXKnkZ4,7292
-reflectorch/data_generation/priors/parametric_models.py,sha256=vaAGcm9Ky-coidNliE4R1YvoqCcbNwEwJVCwwmkKI4E,25139
-reflectorch/data_generation/priors/parametric_subpriors.py,sha256=mN4MELBOQurNponknB_1n46fiNncXkI5PnnPL3hJee4,14352
+reflectorch/data_generation/priors/parametric_models.py,sha256=RBB67T5skHyC6ZlVtzRwuTis8VsCwFyIOoe5OsfMiu4,29300
+reflectorch/data_generation/priors/parametric_subpriors.py,sha256=KPW3Y12A5ba405ofGdGny_FIzoFMd90IkQcvVcYhpxE,15164
 reflectorch/data_generation/priors/params.py,sha256=JpH-LRxTztc0QP-3QwjLdu0MsVc1rSixtcscYGs_2Ew,8238
-reflectorch/data_generation/priors/sampler_strategies.py,sha256=U-v5dXpFLJq939aQKvNl7n1Lih-y97K8WJCXBFRQiA0,13059
+reflectorch/data_generation/priors/sampler_strategies.py,sha256=ls9NRwjIsQJlgqv_m9l9MVm3tu4Ivafq5OzG2Q0sI3w,15791
 reflectorch/data_generation/priors/scaler_mixin.py,sha256=gI64v2KOZugSJWaLKASR34fn6qVFl-aoeVA1BR5yXNg,2648
 reflectorch/data_generation/priors/subprior_sampler.py,sha256=TE8DOQhxVr69VmWSjwHyElpgVjOhZKBIPChFWNVYzRc,14769
 reflectorch/data_generation/priors/utils.py,sha256=bmIZYHq95gG68kETfNH6ygR39oitUEJ0eCO_Fb8maH0,3836
-reflectorch/data_generation/reflectivity/__init__.py,sha256=x8XivTg1ygz1Ky_N1eVvDfz2KS-9j9U2px0kQM3Bgf4,3145
-reflectorch/data_generation/reflectivity/abeles.py,sha256=3NXGXaBiOyAiFfy3Za7WATNLhTpKrNzieYdCsmljvQ8,2625
-reflectorch/data_generation/reflectivity/kinematical.py,sha256=QLX3erfSmEwP2n_x8gQlfaGr3pNAxcJ95Efx4Uu2CKk,2626
-reflectorch/data_generation/reflectivity/memory_eff.py,sha256=iIufbdEJGv9nc-Sr51gFQHEYVdgmbrAD-F3ydNE30nU,4024
+reflectorch/data_generation/reflectivity/__init__.py,sha256=ryblJgCvnjBVPLpmOVovtZtdkEfJ_H26dqthdHfIqQ4,5209
+reflectorch/data_generation/reflectivity/abeles.py,sha256=Pcov-m0HImw1DOR57t0OsBOLZs3MA6fk7RePf_VCEZU,3097
+reflectorch/data_generation/reflectivity/kinematical.py,sha256=AKZqdA4dRU6Gvs7sq6ODLToMmImrMZWMCOMrzpoX2D4,2639
+reflectorch/data_generation/reflectivity/memory_eff.py,sha256=pbsXkcGwDorey6qVx2SlBK6IlzNJY1FeeF84cEJB_hY,4023
 reflectorch/data_generation/reflectivity/numpy_implementations.py,sha256=QBzn4yVnOdlkHeeR-ZFPS115GnLdO9lMTGO2d3YhG9I,3177
-reflectorch/data_generation/reflectivity/smearing.py,sha256=pc95Lig9NIWtHDCKnKbLp7G_kmL7_3YB6ZpDphgC2D8,4001
+reflectorch/data_generation/reflectivity/smearing.py,sha256=qxaEi43Wv9OZHRQr-nQJbtPJEIaXb2NEA1q3evRhCCc,4395
+reflectorch/data_generation/reflectivity/smearing_pointwise.py,sha256=y3iSc0wKh6wF4xzD24Zz2OmerwV8HKQZAg0Y3kpzPuU,3688
 reflectorch/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reflectorch/extensions/jupyter/__init__.py,sha256=bs6RQQ8LRcKAJu2I9K8BZEMQSll3kLdirRvxLbgsY9w,72
 reflectorch/extensions/jupyter/callbacks.py,sha256=K4Z8Mlm_Pa80YDsNOdMZLIJZyVrg09kBVzVdFHKgQFk,1074
 reflectorch/extensions/matplotlib/__init__.py,sha256=ARmc228xbZUj4fWNq5fVxZ8PN444fjDNxNP4ytVCyfo,104
 reflectorch/extensions/matplotlib/losses.py,sha256=H2vrjNqzmQ0BP4377uN4uhlH88aS5OAfA7pwRM2WHqA,686
+reflectorch/extensions/refnx/refnx_conversion.py,sha256=XBHT2zL9XOgL5h2Q-_yDJScUBAsBd0xlivXEMeVGmao,3450
 reflectorch/inference/__init__.py,sha256=i0KNn83XN33mLrV7bpHdLd0SXxuGCKfbQcIoa247Uts,834
-reflectorch/inference/inference_model.py,sha256=aVAZkTfVe2cweitcVVyVuqDsb6FkLbh4Wun4_K_MkTM,36599
+reflectorch/inference/inference_model.py,sha256=1OBZA00xXNchFKxYxpaPFLdFPTcQYlxR3h7q7wSK9WE,44243
 reflectorch/inference/multilayer_fitter.py,sha256=0CxDpLOEp1terR4N39yFlxhvA8qAbHf_01NbmvYadck,5510
 reflectorch/inference/multilayer_inference_model.py,sha256=hH_-dJGdMOox8GHXdM_nODXDlNgh_v449xW5FmklRdo,7575
+reflectorch/inference/plotting.py,sha256=zycWd2j9soe1NwoIE4X1iiGV-ghHH3M9TnFCsKN2jEQ,3868
 reflectorch/inference/query_matcher.py,sha256=Dk49dW0XreeCjufzYBTKchfTdVbG6759ryV6I-wQL60,3387
 reflectorch/inference/record_time.py,sha256=3er-aoR8Sd_Kc4qNwUmRqkEz4FYhVxdi1ARnBohybzM,1140
 reflectorch/inference/sampler_solution.py,sha256=DeJM3EXEb6S5EiASj3mmdNI-Y07Cr5UzzA5oq-vEB-Q,2288
-reflectorch/inference/scipy_fitter.py,sha256=339M33OdmfgOpifJGLYk4KVcnnNJrY6_aH7Lz6Vtt24,5404
+reflectorch/inference/scipy_fitter.py,sha256=5xUZ1JDRXbdQF1yZotMaDFQnxYioqycoZknCWyGvZOY,8201
 reflectorch/inference/torch_fitter.py,sha256=j1NzkzLCmQ4H6LfIi82LsSBmIdunnWzm3kbGx-hqvDs,3391
 reflectorch/inference/preprocess_exp/__init__.py,sha256=bR6H-xgBD96z4P9P1T2ECnWvalrimdMTfTNArIWPLy0,383
 reflectorch/inference/preprocess_exp/attenuation.py,sha256=UKDfUjCKKMgAuEs5Ttyo0KEQmpvHZI52UgVflh7T81A,1518
@@ -54,30 +57,30 @@ reflectorch/inference/preprocess_exp/footprint.py,sha256=xc409M5X-QW0Ce_6dEZdj8N
 reflectorch/inference/preprocess_exp/interpolation.py,sha256=o2v-mlfRYzeaaikeQVeY7EA7j-K42dMfT12oN3mk51k,694
 reflectorch/inference/preprocess_exp/normalize.py,sha256=09v7nZdtw6SnW_67xFrPnqzOA5AtFBGarjK4Pfn4VoE,695
 reflectorch/inference/preprocess_exp/preprocess.py,sha256=pyyq8fSvcm1bWAShzGHYnKOc55Rofh4FIw1AC7Smq-U,5111
-reflectorch/ml/__init__.py,sha256=TJERkE_itNOH3GwtC-9Xv0rZ70qEkMukFLD1qXsMdOQ,730
-reflectorch/ml/basic_trainer.py,sha256=2z_3Iy_9dEgOha8da69RjiMQ_89C2SoJM9omh0WO-ek,9647
-reflectorch/ml/callbacks.py,sha256=blZNFubtRQkcx5sQrTNkARiPj75T8VdCn6CYJNN7hKg,2743
+reflectorch/ml/__init__.py,sha256=0LVWweeidkoZLqbDYKi-wx5BvG7DYwsR9IRKM3TZoOM,790
+reflectorch/ml/basic_trainer.py,sha256=GbSs4ie5yzY7nhGiqHO5D4HZCqB-taKenaswf1dIvok,10073
+reflectorch/ml/callbacks.py,sha256=u9jNvxVEIA0a_SpmT-gnP8qoC6-C8p0F4TvIuBQr-ek,2801
 reflectorch/ml/dataloaders.py,sha256=IvKmsH5gX_b-00KRFeL-x3keEfBcvYkQFWGWd8Caj-I,1073
-reflectorch/ml/loggers.py,sha256=oYZskGMbbWfW3sOXOC-4F-DIL-cLhYrRFeggrmPtPGM,743
-reflectorch/ml/schedulers.py,sha256=AO-dS1bZgn7p-gGJJhocmL6Vc8XLMwynZxvko4BCEsw,10048
-reflectorch/ml/trainers.py,sha256=DXO0_ue3iVQKi8vDttEQZbNLcn3xwkRBIFrw0XTomn4,3643
+reflectorch/ml/loggers.py,sha256=Dvjte9rmDxoA8AB_7cFWWxkvaQBdhn2_ag2jYCM0uSQ,1478
+reflectorch/ml/schedulers.py,sha256=p_Z-TqGsu45KencD_EwX6hGMMsvvTUnQtkwC98kKTDI,14227
+reflectorch/ml/trainers.py,sha256=-1I4asSg2hYvsT6w5yru2RNG59OZWnykANTlAiHpubg,8142
 reflectorch/ml/utils.py,sha256=_OKkc6o5od7vnfBNZKWsGApJxA62TrAlLok92s0nG4k,71
-reflectorch/models/__init__.py,sha256=vsGXi1BoLVyki_OF_Ezv9GXCpuJ22liGwolVZ0rxyyM,335
+reflectorch/models/__init__.py,sha256=jJUUjyFJBIZHXqCeaxxnlazDLAE5VbPrdYUYRAI95lo,345
 reflectorch/models/activations.py,sha256=LDiIxCnLFb8r_TRBZSt6vdOZmexCWAGa5DfE_SotUL8,1431
-reflectorch/models/encoders/__init__.py,sha256=X9cHeWjJGVXNFwii9ZJasNbqrV4kt9QyJYPdKfxgp04,443
-reflectorch/models/encoders/conv_encoder.py,sha256=Xa9Yo2lDctFWlnz8vIIdzwakcE_cgBR_J92UL59XhbA,7653
+reflectorch/models/encoders/__init__.py,sha256=_cGxemTtkDZa5SoCGiz-1o9J1J4FyuRAqH_PyTmiXVI,413
+reflectorch/models/encoders/conv_encoder.py,sha256=LLvqnt268VR6C_1UDR24qka_9ijHc1hvvciJaV8AmQo,7910
 reflectorch/models/encoders/conv_res_net.py,sha256=VP1CCKdNqsva2R4l-IV6d63mQk6bI4Aresfd6b6_YKU,3343
-reflectorch/models/encoders/fno.py,sha256=s_S7hnpLE7iGfyvnQ-QvTh0rKO5KFiy5tUYau4sJbvI,4693
-reflectorch/models/networks/__init__.py,sha256=zHUvrlb0KVOpRrwVjwjR1g8sVWX6VH80FLGqu4jiux8,291
-reflectorch/models/networks/mlp_networks.py,sha256=C7py6qCBVaYYt0FMEf8gbT4lndArKpUYYgTN1001-T8,11614
-reflectorch/models/networks/residual_net.py,sha256=msDJaDw7qL9ebEW1Avw6Qw0lgni68AMgF4kXiJKzeaQ,4637
+reflectorch/models/encoders/fno.py,sha256=G1Iav7IYVdvogK40dlKK1zsCTRIkesffrcVTywRZKjI,5115
+reflectorch/models/networks/__init__.py,sha256=QeNcWShZ7i-hoEFOLnOoq7U6WzBoXj7UIHCCJnZ99KA,341
+reflectorch/models/networks/mlp_networks.py,sha256=5Sr6Rg4uZca5ghj9mJAHMtB9MXAcdGUIAR8EHU9MdH4,20441
+reflectorch/models/networks/residual_net.py,sha256=oEPJTWjvKO1yniLTT9B54Z_ZbZv4i1uHsP9ongbn-ko,5643
 reflectorch/runs/__init__.py,sha256=xjuZGjqZuEovlpe9Jj5d8Nn5ii-5jvAYdeHT5oYaYVI,723
 reflectorch/runs/config.py,sha256=8YtUOXr_DvNvgpu59CNAr3KrQijx1AGDE95gYrsuwsM,804
 reflectorch/runs/slurm_utils.py,sha256=Zyj4_K5YpiWNJhgpFLWYHsSaaI-mgVEWsN15Gd_BhI0,2600
-reflectorch/runs/train.py,sha256=e-Jj0fwYlUB2NLDxCzy0cLsSrJbNqd5pN5T1L7-Eiig,2560
-reflectorch/runs/utils.py,sha256=bp3Nwd6pfX5yFlTVX28zDtSTo2gg2JZh9HCpZpvnJWk,12637
-reflectorch-1.2.0.dist-info/LICENSE.txt,sha256=2kX9kLKiIRiQRqUXwk3J-Ba3fqmztNu8ORskLBlAuKM,1098
-reflectorch-1.2.0.dist-info/METADATA,sha256=dulVsQJvbWfaEPJ71tgp_CVS1AwUnba2Y-Wph3c1pRk,7805
-reflectorch-1.2.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-reflectorch-1.2.0.dist-info/top_level.txt,sha256=2EyIWrt4SeZ3hNadLXvEVpPFhyoZ4An7YflP4y_E3Fc,12
-reflectorch-1.2.0.dist-info/RECORD,,
+reflectorch/runs/train.py,sha256=7BLjKjI3H2ZwqMhmlO4ebsz6X804e_O9XU73xKyOJW0,2503
+reflectorch/runs/utils.py,sha256=maeHHXK3CdEnewPZTWpDO6lr_a8gwEyup6aSLGw_MGM,14055
+reflectorch-1.3.0.dist-info/licenses/LICENSE.txt,sha256=2kX9kLKiIRiQRqUXwk3J-Ba3fqmztNu8ORskLBlAuKM,1098
+reflectorch-1.3.0.dist-info/METADATA,sha256=rFJSL1nVJTbJu0Fo1DenYTpRmLbILtf-I5SLXq8ExaQ,7800
+reflectorch-1.3.0.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+reflectorch-1.3.0.dist-info/top_level.txt,sha256=2EyIWrt4SeZ3hNadLXvEVpPFhyoZ4An7YflP4y_E3Fc,12
+reflectorch-1.3.0.dist-info/RECORD,,

{reflectorch-1.2.0.dist-info → reflectorch-1.3.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any