Spaces:
Runtime error
Runtime error
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" | |