Welcome to the SuperField Network Training and Inference Code repository. This guide will help you set up and run the code effectively.
Make sure all the packages listed in requirements.txt
are installed. You can do this by running:
pip install -r requirements.txt
To start training the SuperField Network from scratch, use the train.py
script:
python train.py
For out-of-the-box SuperField training, navigate to the Inference
folder and run the Infer.py
script:
python Inference/Generate_SuperField.py
To use pretrained models, make sure to download them from the shared link and place them in the appropriate directory.
Feel free to open an issue or send me a message (atapp@childrensnational.org) if you encounter any problems or have questions!