Spaces:
Runtime error
Runtime error
File size: 334 Bytes
3e99b05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from .conditional_detr_r50_50ep import (
train,
dataloader,
optimizer,
lr_multiplier,
model,
)
# modify model config
model.backbone.stages.depth = 101
# modify training config
train.init_checkpoint = "detectron2://ImageNetPretrained/torchvision/R-50.pkl"
train.output_dir = "./output/conditional_detr_r101_50ep"
|