(debug) import yaml

Update model card

Git lfs changed to accept png and jpg

(minor) push paper files for easier LLM usage

Add final pt to git

Safetensors model export

Safetensors export and infer.

Add images to README

fix strip checkpoint saving

README and cleanup

Metrics computation in infer.py

(minor util) strip checkpoint util

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

Attempting export to onnx and tensorRT

(test) test ResNet backbone

Add export to onnx and tensorRT

Add benchmark to infer.py

(debug) fix eval loop not selecting randomly

Save test visuals during final eval

Add ResNet-50 as a backbone option

(cleanup) removing timm from the codebase.

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

(debug) to torch operations in bfloat conversions.

Add final checkpoint save and test set val

(untested test) fixing some test errors

Eval to infer.py

(optim) benchmark=True and higher eval batch size

Speeding up eval by reducing iters

Added precision training option

(optim) prefetch and speed up eval.

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

to mit_b2 as per paper

(debug) eval fixes

(devops) automatic pull and preprocess of datasets

(config) reducing training iters and eval, checkpoint intervals

(test) test dataset

Pull ttpla script

Running training

(format) format

Training loop and around the codebase

Dataset pull script

Project skeleton generated

Unpack `.tex` source.

Repo init (paper add)