ForkedHulk2 / experiments /release /vd_par_hm36_flip_test.yaml
tuandunghcmut's picture
Upload folder using huggingface_hub
345ee20 verified
raw
history blame
995 Bytes
common:
label_proj:
type: dense_labeling_projector
kwargs: # kept one
mask_times_cls: True
backbone:
kwargs:
test_pos_mode: False
mask_all_gt_tokens: True
tester:
type: TesterMAEDev
use_ceph: True
sync: True
collate: dev
model_entry_kwargs:
test_flag: par_flip
flip_channels: [[1, 4],
[2,5],
[3,6],
[7,10],
[8,11],
[9,12],
[14,16],
[15,17],
[21,23],
[22,24]]
tasks:
0:
name: ParTest
loss_weight: 1.0 # *redundant*
gres_ratio: 1 # int, > 0| world/sum*ratio, *redundant*
dataset:
type: Human3M6ParsingDataset
kwargs:
dataset: val
is_train: False
cfg:
eval_crop_size: [ 480, 480 ]
is_flip: False
is_multi_scale: False
sampler:
batch_size: 16 # per card