Skip to content

Add the torch pipeline

Samuel MICHEL requested to merge add_torch_pipeline into main
  • copy and adapt predictor and trainer for torch model
  • adapt train,predict,analyze and experiment script to be compatible with either scikit and torch pipeline
  • add documentation and tests
  • fix resampling process and transform (RawToEpochs)
  • add resources, summary, model checkpointer
  • add saving and loading prediction in hdf5 format for torch and scikit pipeline
Edited by Samuel MICHEL

Merge request reports

Loading