jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
35 lines
655 B
YAML
35 lines
655 B
YAML
image_root: '/export/share/datasets/vision/coco/images/'
|
|
ann_root: 'annotation'
|
|
dataset: 'coco'
|
|
|
|
# set pretrained as a file path or an url
|
|
pretrained: 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_retrieval_coco.pth'
|
|
|
|
# size of vit model; base or large
|
|
|
|
vit: 'base'
|
|
batch_size_train: 32
|
|
batch_size_test: 64
|
|
vit_grad_ckpt: True
|
|
vit_ckpt_layer: 4
|
|
init_lr: 1e-5
|
|
|
|
# vit: 'large'
|
|
# batch_size_train: 16
|
|
# batch_size_test: 32
|
|
# vit_grad_ckpt: True
|
|
# vit_ckpt_layer: 12
|
|
# init_lr: 5e-6
|
|
|
|
image_size: 384
|
|
queue_size: 57600
|
|
alpha: 0.4
|
|
k_test: 256
|
|
negative_all_rank: True
|
|
|
|
# optimizer
|
|
weight_decay: 0.05
|
|
min_lr: 0
|
|
max_epoch: 6
|
|
|