Skip to content

Version 2.3

Latest
Latest
Compare
Choose a tag to compare
@lhmbev lhmbev released this 01 Sep 12:54
· 2 commits to master since this release
v2.3
fe80a06

Adds the ability to train across multiple nodes with multiple GPUs, some Jupyter Notebooks to analyse results, and some minor bug fixes/clean-ups.

Changes

  • Added the ability to train using PyTorch's DistributedDataParallel module, speeding up training by around 5x
  • Changed Generator architecture to only use 4x4 kernels
  • Added more control over saving a model during training; an interval in epochs can be specific at which to save the model
  • Added lists of clean and stripe indices to PatchVisualiser class
  • Models are now saved with the --name argument, rather than the --mode argument
  • Added parameter for the Centre of Rotation in the reconstruct() function
  • Added a script to generate a mask of stripe locations
  • Added a script to apply the model to arbitrary tomographic scans
  • Added Jupyter Notebooks for visualising results, RMSEs, residuals and graphs
  • Added a tutorial on how to use the repository
  • Updated documentation and removed old/unused code
  • Minor bug fixes
Assets 2