edflow.main module

Summary

Functions:

test	Run tests.
train	Run training.

Reference

edflow.main.train(config, root, checkpoint=None, retrain=False, debug=False)

Run training. Loads model, iterator and dataset according to config.

edflow.main.test(config, root, checkpoint=None, nogpu=False, bar_position=0, debug=False)

Run tests. Loads model, iterator and dataset from config.