wehub-resource-sync
16031aae96
CPU tests Workflow / Testing (ubuntu-latest, 3.12) (push) Failing after 1s
CPU tests Workflow / Testing (ubuntu-latest, 3.13) (push) Failing after 0s
Mypy Type Check / Type Check (push) Failing after 0s
Docs/Test WorkFlow / Test docs build (push) Failing after 1s
PR Conflict Labeler / labeling (push) Failing after 1s
Dependency resolution / Resolve [tflite] extra — Python 3.12 (push) Failing after 0s
Smoke Tests / try-all-models (ubuntu-latest, 3.10) (push) Failing after 0s
Smoke Tests / try-all-models (ubuntu-latest, 3.13) (push) Failing after 1s
CPU tests Workflow / build-pkg (push) Failing after 1s
CPU tests Workflow / Testing (ubuntu-latest, 3.10) (push) Failing after 0s
CPU tests Workflow / Testing (ubuntu-latest, 3.11) (push) Failing after 0s
Smoke Tests / try-all-models (macos-latest, 3.10) (push) Has been cancelled
Smoke Tests / try-all-models (macos-latest, 3.13) (push) Has been cancelled
Smoke Tests / try-all-models (windows-latest, 3.10) (push) Has been cancelled
Smoke Tests / try-all-models (windows-latest, 3.13) (push) Has been cancelled
CPU tests Workflow / Testing (macos-latest, 3.10) (push) Has been cancelled
CPU tests Workflow / Testing (macos-latest, 3.13) (push) Has been cancelled
CPU tests Workflow / Testing (windows-latest, 3.10) (push) Has been cancelled
CPU tests Workflow / Testing (windows-latest, 3.13) (push) Has been cancelled
CPU tests Workflow / testing-guardian (push) Has been cancelled
GPU tests Workflow / Testing (push) Has been cancelled
22 lines
701 B
YAML
22 lines
701 B
YAML
# RF-DETR Medium — example training configuration (resolution 576, patch 16).
|
|
# Usage:
|
|
# rfdetr fit --config configs/rfdetr_medium.yaml
|
|
# rfdetr fit --config configs/rfdetr_medium.yaml \
|
|
# --model.train_config.init_args.dataset_dir /data/my_dataset \
|
|
# --trainer.devices 4
|
|
|
|
model:
|
|
model_config:
|
|
class_path: rfdetr.config.RFDETRMediumConfig
|
|
init_args:
|
|
num_classes: 80 # set to your dataset class count
|
|
train_config:
|
|
class_path: rfdetr.config.TrainConfig
|
|
init_args:
|
|
dataset_dir: /data/coco # required: path to your COCO-format dataset
|
|
output_dir: output/rfdetr_medium
|
|
epochs: 100
|
|
batch_size: 4
|
|
num_workers: 4
|
|
tensorboard: true
|