ECOC: Error-Correcting Output Codes
python ./pytorch/train.py --model-name lenet5 --lr 0.001 --codebook_name=hunqun_deng_c10_n5 --do-train --do-eval --epochs 400 --batch-size 32 --loss binary_cross_entropy
# ResNet18
python ./pytorch/train.py --model-name resnet18 --lr 0.1 --codebook_name=hunqun_deng_c10_n5 --do-train --do-eval --epochs 200 --batch-size 32 --loss binary_cross_entropy
python ./pytorch/train.py --codebook_name=hunqun_deng_c10_n5 --do-train --do-eval --epochs 200 --batch-size 32 --loss binary_cross_entropy --model-path ./output/ECOC-LeNet-5_MNIST_sgd_20230520-020335/ECOC-LeNet-5.pt
# LeNet5
python ./pytorch/train.py --do-train --do-eval --epochs 200 --batch-size 32 --loss cross_entropy
# ResNet18
python ./pytorch/train.py --model-name resnet18 --lr 0.1 --do-train --do-eval --epochs 200 --batch-size 32 --loss cross_entropy
For ModuleNotFoundError: No module named 'ecoc'
or ModuleNotFoundError: No module named 'pytorch'
etc, add '.' to PYTHONPATH:
export PYTHONPATH=.:PYTHONPATH
Windows:
set PYTHONPATH=%PYTHONPATH%;.
See examples/code_sets.md
Date | log |
---|---|
Jun 20, 2020 | Initially implemented ECOC generation codes. |
Thanks to Dr. Yunxiang Yao's (from ECE Dept. of HKUST) explaination on correcting codes.