λ§μ½ μλ‘μ΄ κ²½λν ꡬ쑰μ λͺ¨λΈμ μ°Ύμλ΄μ μλΉμ€κΉμ§ ν΄λ³΄κ³ μΆμΌμλ€λ©΄, μ°λ¦¬ νλ«νΌ
μ μ¬μ©ν΄λ³΄μΈμ.
$> tree -d
.
βββ configs
β βββ model
βββ src
β βββ augmentation
β βββ modules
β βββ utils
β βββ custom_dataset.py
β βββ dataloader.py
β βββ decomp.py
β βββ loss.py
β βββ model.py
β βββ network_prune.py
β βββ trainer.py
β βββ vbmf.py
βββ model_decomp.py
βββ model_search.py
βββ requirement.txt
βββ torch2tflite.py
- Pytorch
- TorchVision
- optuna
- ....
NASλ₯Ό ν΅ν΄ κ²½λν λͺ¨λΈμ μ°Ύμ μ μμ΅λλ€.
Args μμ | μ€λͺ |
---|---|
METRIC | λͺ¨λΈμ μ±λ₯μ νκ°νλ μ§νμ λλ€. [F1, ACC] |
LIMIT_MACS | μ°Ύκ³ μνλ λͺ¨λΈμ MACSμ μ νμ κ±Έμ΄ μ€λλ€. |
image_size | νκΉμ΄ λλ λ°μ΄ν°μ μ΄λ―Έμ§ μ¬μ΄μ¦λ₯Ό μ μν©λλ€. |
batch_size | dataloaderμ λ°°μΉμ¬μ΄μ¦λ₯Ό μ§μ ν΄μ€λλ€. |
CLASSES | λΆλ₯νκ³ μνλ λ°μ΄ν°μ ν΄λμ€ μλ₯Ό μ§μ ν©λλ€. |
MAX_DEPTH | μ°Ύκ³ μνλ λͺ¨λΈμ μ΅λ κΉμ΄λ₯Ό μ§μ ν΄μ£ΌμΈμ. |
data_type | λ°μ΄ν° μ νμ μ§μ ν΄μ€λλ€. [CIFAR10, CIFAR100, CUSTOM] |
data_root | λ°μ΄ν°κ° μ μ₯λ λλ ν°λ¦¬ κ²½λ‘λ₯Ό μ§μ ν©λλ€. |
Trial | λͺ©ν μν€ν μ³λ₯Ό μ°ΎκΈ° μν΄ λͺλ²μ Trialμ ν κ²μΈμ§ μ§μ ν©λλ€. |
prun_type | μν€ν μ³λ₯Ό μ°Ύμ λ μ΄λ€μμΌλ‘ λͺ¨λΈμ μ λ³νμ¬ κ°μ§κ³ μ¬κ²μΈμ§ μ ν΄μ€λλ€. [0.μ¬μ©μν¨, 1.optunaμ λ΄μ₯λ prunerλ₯Ό μ¬μ©, 2. custom] |
auto_augment | μ€ν μ΄κ·Έλ©ν μ΄μ μ μ¬μ©ν μ§μ λν μ¬λΆλ₯Ό μλ €μ€λλ€. |
NASλ₯Ό ν΅ν΄ μ°Ύμ λͺ¨λΈμ μ’ λ κ°λ³κ² ν΄μ£ΌκΈ° μν΄ decompositionλ¨κ³λ₯Ό κ±°μΉ©λλ€.
Pytorchλͺ¨λΈμ λ€μν νλ«νΌμμ μλΉμ€ νκΈ°μν΄ μ¬λ¬ νλ«νΌμΌλ‘ λ³νν΄μ£Όλ μμ
μ ν΄μ€λλ€.
ν΄λΉ νλ‘μ νΈλ₯Ό ν΅ν΄ GTSNetμ΄λΌλ λͺ¨λΈμ μ°Ύμ νμ΅μν¨ν μ¬λ¬ νλ‘μ νΈμ μ μ©μ ν΄λ³΄μμ΅λλ€.
- Android Project: https://github.com/gihop/MaskCheckIn-boostcamp-p4-GTS-side-project
- Raspberry Pi Project: https://github.com/happyBeagle/rpi_image