Fine-tuning
davidmoralesrodriguez opened this issue · comments
davidmoralesrodriguez commented
Hi, is it possible to fine-tune a model on a custom dataset using the pretrained imagenet models?
YuanLi commented
Of course, it can.
Simply, you can:
from timm.models import create_model
model = = create_model(
args.model,
pretrained=args.pretrained,
num_classes=args.num_classes,
drop_rate=args.drop,
drop_connect_rate=args.drop_connect, # DEPRECATED, use drop_path
drop_path_rate=args.drop_path,
drop_block_rate=args.drop_block,
global_pool=args.gp,
bn_tf=args.bn_tf,
bn_momentum=args.bn_momentum,
bn_eps=args.bn_eps,
checkpoint_path=args.initial_checkpoint,
img_size=args.img_size)
I will update the codes of how to finetune on other datasets like CIFAR10/100 recently.