Spaces:
Runtime error
Runtime error
File size: 761 Bytes
3e99b05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from detrex.config import get_config
from .dino_focalnet_large_lrf_384_fl4_5scale_12ep import (
train,
optimizer,
model,
)
EPOCH_ITERS = 4264
dataloader = get_config("common/data/odor_demo.py").dataloader
model.num_classes = 139
# using 36ep scheduler
lr_multiplier = get_config("common/coco_schedule.py").lr_multiplier_36ep
# modify training config
train.max_iter = 50 * EPOCH_ITERS
train.init_checkpoint = "pretrained-models/focalnet_large_lrf_384_fl4.pth"
train.output_dir = "detrex-output/odor3-tests"
train.eval_period = EPOCH_ITERS
train.checkpointer.period = EPOCH_ITERS
train.final_checkpoint = "trained-models/dino-focal/focaldino_ep18.pth"
# using larger drop-path rate for longer training times
model.backbone.drop_path_rate = 0.4
|