Skip to content

Latest commit

 

History

History
20 lines (12 loc) · 593 Bytes

README.md

File metadata and controls

20 lines (12 loc) · 593 Bytes
Raw

DON_HH

Learning Hodgkin-Huxley model with DeepONet (DON), Fourier Neural Operator (FNO) and Wavelets Neural Operator (WNO).

File and folders:

  • deepONet_HH_pytorch.py : main
  • default_params_new.yml : config yml file template
  • src : local modules for DeepONet_HH_pytorch.py
  • legacy_files : old files and prototipes

Usage:

python3 deepONet_HH_pytorch.py --config_file <yml_file.yml>

To recover trained models:

python3 recover_model.py --config_file <yml_file.yml>

please change lines in the code indicating the new indexes to predict (idx) and the model path (modelname)