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