WireSegHR / train.py

Commit History

README and cleanup
e05cc17

MRiabov commited on

(minor refactor) os.path to pathlib and tdqm to infer.py
8223f36

MRiabov commited on

(debug) fix eval loop not selecting randomly
32c94fb

MRiabov commited on

Save test visuals during final eval
ba1bfde

MRiabov commited on

Merge branch 'main' of https://github.com/MRiabov/wire-seg-hr-impl
8cd5198

MRiabov commited on

(debug) to torch operations in bfloat conversions.
854831b

MRiabov commited on

Add final checkpoint save and test set val
feec877

MRiabov commited on

Eval to infer.py
500cad9

MRiabov commited on

(optim) benchmark=True and higher eval batch size
d2db539

MRiabov commited on

Speeding up eval by reducing iters
8b1ceda

MRiabov commited on

Added precision training option
8743e1d

MRiabov commited on

(optim) prefetch and speed up eval.
efdb6a6

MRiabov commited on

(debug) 6 chan instead of 7. (warn: conflicting git changes)
a527623

MRiabov commited on

to mit_b2 as per paper
d46d294

MRiabov commited on

(debug) eval fixes
e160020

MRiabov commited on

(devops) automatic pull and preprocess of datasets
953508f

MRiabov commited on

Running training
61275b8

MRiabov commited on