training_args.json

{
  "cmdline": "train_detection.py --network deformable_detr_boxref --tag coco --backbone convnext_v2_tiny --backbone-tag imagenet21k --backbone-pretrained --opt adamw --lr 0.0002 --backbone-lr 0.00002 --lr-scheduler step --lr-step-size 40 --lr-step-gamma 0.1 --batch-size 2 --epochs 50 --wd 0.0001 --grad-accum-steps 16 --multiscale --max-size 1152 --clip-grad-norm 0.1 --amp --amp-dtype bfloat16 --compile-backbone --compile-opt --save-frequency 1 --data-path ~/Datasets/cocodataset/train2017 --val-path ~/Datasets/cocodataset/val2017 --coco-json-path ~/Datasets/cocodataset/annotations/instances_train2017.json --coco-val-json-path ~/Datasets/cocodataset/annotations/instances_val2017.json --class-file public_datasets_metadata/coco-classes.txt --no-summary",
  "network": "deformable_detr_boxref",
  "net_param": null,
  "model_config": null,
  "backbone": "convnext_v2_tiny",
  "backbone_param": null,
  "backbone_model_config": null,
  "backbone_tag": "imagenet21k",
  "backbone_epoch": null,
  "backbone_pretrained": true,
  "pretrained": false,
  "reset_head": false,
  "freeze_body": false,
  "freeze_backbone": false,
  "freeze_backbone_stages": null,
  "compile": false,
  "compile_backbone": true,
  "compile_dynamic": false,
  "compile_opt": true,
  "opt": "adamw",
  "momentum": 0.9,
  "nesterov": false,
  "opt_eps": null,
  "opt_betas": null,
  "opt_alpha": null,
  "lr": 0.0002,
  "bias_lr": null,
  "backbone_lr": 2e-05,
  "lr_scale": null,
  "lr_scale_type": "linear",
  "wd": 0.0001,
  "norm_wd": null,
  "bias_weight_decay": null,
  "transformer_embedding_decay": null,
  "layer_decay": null,
  "lr_scheduler_update": "epoch",
  "lr_scheduler": "step",
  "lr_step_size": 40,
  "lr_steps": null,
  "lr_step_gamma": 0.1,
  "lr_cosine_min": 1e-06,
  "lr_power": 1.0,
  "grad_accum_steps": 16,
  "channels": 3,
  "size": [
    640,
    640
  ],
  "max_size": 1152,
  "dynamic_size": false,
  "multiscale": true,
  "freeze_backbone_bn": false,
  "sync_bn": false,
  "batch_size": 2,
  "warmup_epochs": 0,
  "aug_type": "birder",
  "aug_level": 4,
  "rgb_mode": "birder",
  "epochs": 50,
  "stop_epoch": 51,
  "save_frequency": 1,
  "keep_last": null,
  "resume_epoch": null,
  "load_states": false,
  "load_scheduler": false,
  "model_ema": false,
  "model_ema_steps": 1,
  "model_ema_decay": 0.9998,
  "ra_sampler": false,
  "ra_reps": 3,
  "tag": "coco",
  "log_interval": 20,
  "num_workers": 8,
  "prefetch_factor": null,
  "amp": true,
  "amp_dtype": "bfloat16",
  "fast_matmul": false,
  "grad_anomaly_detection": false,
  "world_size": 2,
  "dist_url": "env://",
  "clip_grad_norm": 0.1,
  "gpu": 0,
  "cpu": false,
  "use_deterministic_algorithms": false,
  "binary_mode": false,
  "plot_lr": false,
  "no_summary": true,
  "val_path": "~/Datasets/cocodataset/val2017",
  "data_path": "~/Datasets/cocodataset/train2017",
  "coco_val_json_path": "~/Datasets/cocodataset/annotations/instances_val2017.json",
  "coco_json_path": "~/Datasets/cocodataset/annotations/instances_train2017.json",
  "class_file": "public_datasets_metadata/coco-classes.txt",
  "ignore_file": null,
  "rank": 0,
  "distributed": true,
  "dist_backend": "nccl"
}