Logo, pydata sphinx theme, custom state loading, share train and val states

New Features

• Share train and val states between phases with phase_states attribute of state.

   @argus.callbacks.on_epoch_complete
   def some_validation_callback(state: argus.engine.State):
       train_step_output = state.phase_states['train'].step_output
       ...

• Option to use custom state load function for argus.load_model.

  def state_load_from_dir(dir_path):
      file_path = pathlib.Path(dir_path) / 'some_model_name.pth'
      return torch.load(dir_path)
  
  model = load_model(path_to_dir_with_model,
                     state_load_func=custom_state_load_func)

Docs

• Argus logo!
  
• Migrate to pydata-sphinx-theme.

Fix

• Fix sdist package installation by adding MANIFEST.in with requirements.txt.

Examples

• Use torch.cuda.amp instead Apex in advanced CIFAR example.
• Add as an example solution for RANZCR CLiP - Catheter and Line Position Challenge.

Chore

• setup.cfg with pytest and flake8 settings.
• CI check code style with flake8.
• Run tests on macOS and Windows.
• Update Dockerfile and tests to PyTorch 1.8.0.
• Update Dockerfile to CUDA 11.1.