nithishbasireddy
/

el-defect-training

Model card Files Files and versions

el-defect-training / train.py

Commit History

Fix: mask dtype → LongTensor for DiceLoss one_hot compatibility

d3f1e7d
verified

nithishbasireddy commited on Apr 23

Fix: total_mem → total_memory, NUM_WORKERS=0 for Windows

2ab499c
verified

nithishbasireddy commited on Apr 23

Upload train.py with huggingface_hub

1bac9b8
verified

nithishbasireddy commited on Apr 23