IBM_AI/vi-opton/opton_cls_train/backup_model.bat

REM @echo off

if "%1"=="" (
    echo Usage: backup_model.bat model_id

) else (

    REM set NET=bvlc_alexnet
    set NET=bvlc_googlenet

    set MODEL_ID="%1"
    set MODEL_DIR="d:\\models\\opton\\%MODEL_ID%"
    echo "%MODEL_DIR%"

    if exist "%MODEL_DIR%" (
        echo [FAIL] Backup model folder already exist: %MODEL_DIR%

    ) else (
        set ITERS=100000
        xcopy "%MODEL_ID%\\prototxt" "%MODEL_DIR%\\prototxt" /E /Y /I
        copy "%MODEL_ID%\\output\\snapshot\\%NET%\\%NET%_iter_%ITERS%.caffemodel" "%MODEL_DIR%\\%NET%_final.caffemodel"
        copy "%MODEL_ID%\\config.py" %MODEL_DIR%
        copy "%MODEL_ID%\\data\\images_mean.npy" %MODEL_DIR%
        type nul > "%MODEL_DIR%\\__init__.py"
        type nul > "%MODEL_DIR%\\%MODEL_ID%.tag"

        echo [SUCCESS] Model backup to %MODEL_DIR%
    )

)