visual-anomaly-detection / avatar_rigging.yaml
ifire's picture
Squash
81685e4
raw
history blame contribute delete
648 Bytes
model:
class_path: anomalib.models.EfficientAd
init_args:
imagenet_dir: avatar_rigging_dataset
teacher_out_channels: 384
model_size: S
lr: 0.0001
weight_decay: 1.0e-05
padding: false
pad_maps: true
data:
class_path: anomalib.data.Folder
init_args:
name: avatar_rigging
root: avatar_rigging_dataset
normal_dir: nominal
abnormal_dir: anomaly
train_batch_size: 1
eval_batch_size: 32
num_workers: 8
task: segmentation
train_transform: null
eval_transform: null
test_split_mode: synthetic
test_split_ratio: 0.2
val_split_mode: same_as_test
val_split_ratio: 0.5