/AI_software

Primary LanguagePython

Training Dataset

  • data to pickle file

  1. image to shape

    python prepare_i2s_dataset.py --clip_model_type RN50 --type train

  2. text to color

    python prepare_i2t_dataset.py --clip_model_type RN50 --type train

Training

nohup python -u train.py --config ./configs/train_t2c.yml > ./logs/{lofg_file_name}.txt &

nohup python -u train.py --config ./configs/train_i2s.yml > ./logs/{lofg_file_name}.txt &

Quantization

  1. i2s

  • python {type}_lighter.py --path {ckpt_model_paht} --type {dynamic}

    python i2s_ligther.py --path /home/guest/gihwan/AI_software/term/runs/Image2shape/20231211105651815/valLoss0.12748355151002624_clipcap_RN50x4_Image2ShapeDataset_epoch91.ckpt --type dynamic