v0.6.5

Changes

Estimators

• Added a Physics-Informed CNN estimator for 1D Turbulence
  • example in pimlturb1d_example.ipynb
  • added a unit test on commit as well
  • Gaussian filter layer now supports 1D

Examples

• New plotting_examples.py that include tutorials on:
  • line_plot
  • slice_plot
  • pdf_plot
  • cdf_plot
  • log_plot
  • model_graph
• Adjusted example formatting to unify them in style
• Reduced the size of the example data by 50%

Plotting

• Fixed: model_graph() - model visualizations are fully operational again
  • it also now supports 1D graphs, as well as 2D and 3D

MLflow

• Added new methods
  • log_model() - corresponding to mlflow.pytorch.log_model()
  • load_model() - corresponding to mlflow.pytorch.load_model()

CLI

• Fixed: sapsan get_examples not copying all examples

Training Backend

• Slowly transitioning away from Catalyst - it is no longer required for the model save/load routines
• Added TorchScript option
• Fixed: .next() for iter() in evaluation module
  • also get_loader_shape call of the loader's next iteration