hparams.yaml

accelerator: null
accumulate_grad_batches: 2
activation: gelu
amp_backend: native
amp_level: O2
auto_lr_find: false
auto_scale_batch_size: false
auto_select_gpus: false
aux_loss: true
batch_size: 21
bbox_loss_coef: 5
benchmark: false
check_val_every_n_epoch: 1
checkpoint_callback: true
clip_max_norm: 0.1
coco_path: /home/padeler/work/datasets/coco_2017
dataset_file: coco
debug: false
dec_arch: detr
dec_layers: 6
default_root_dir: null
deterministic: false
dice_loss_coef: 1
dim_feedforward: 1536
distributed_backend: null
dropout: 0.0
enc_layers: 6
eos_coef: 0.1
epochs: 80
eval: false
fast_dev_run: false
flush_logs_every_n_steps: 100
gpus: 0,
gradient_clip_algorithm: norm
gradient_clip_val: 0.0
hidden_dim: 384
init_weights: null
input_size: !!python/tuple
- 192
- 256
limit_predict_batches: 1.0
limit_test_batches: 1.0
limit_train_batches: 1.0
limit_val_batches: 1.0
log_every_n_steps: 50
log_gpu_memory: null
logger: true
lr: 0.0001
lr_backbone: 1.0e-05
lr_drop: 50
mask_loss_coef: 1
max_epochs: null
max_steps: null
max_time: null
min_epochs: null
min_steps: null
move_metrics_to_cpu: false
multiple_trainloader_mode: max_size_cycle
nheads: 8
num_nodes: 1
num_processes: 1
num_queries: 100
num_sanity_val_steps: 2
num_workers: 16
overfit_batches: 0.0
patch_size: 16
plugins: null
position_embedding: enc_xcit
pre_norm: false
precision: 32
prepare_data_per_node: true
process_position: 0
profiler: null
progress_bar_refresh_rate: null
reload_dataloaders_every_epoch: false
replace_sampler_ddp: true
resume_from_checkpoint: null
scale_factor: 0.3
seed: 42
set_cost_bbox: 5
set_cost_class: 1
stochastic_weight_avg: false
sync_batchnorm: false
terminate_on_nan: false
tpu_cores: null
track_grad_norm: -1
truncated_bptt_steps: null
use_det_bbox: false
val_check_interval: 1.0
vit_arch: xcit_small_12_p16
vit_as_backbone: false
vit_dim: 384
vit_dropout: 0.0
vit_weights: https://dl.fbaipublicfiles.com/xcit/xcit_small_12_cp16_dino.pth
weight_decay: 0.0001
weights_save_path: null
weights_summary: top
with_lpi: false