Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How could I train a model from scratch? #3

Open
kvv2018 opened this issue Aug 1, 2018 · 1 comment
Open

How could I train a model from scratch? #3

kvv2018 opened this issue Aug 1, 2018 · 1 comment

Comments

@kvv2018
Copy link

kvv2018 commented Aug 1, 2018

Because my dataset is very different with imagenet and coco. I also want to design a new backbone model that did not provide pre-trained weight. So, I want to train the project from scratch? Is this possible?

if args.params.endswith(".params"):
        net.load_params(args.params)
    elif args.params.endswith(".weights"):
        X = nd.uniform(shape=(1, 3, input_dim, input_dim), ctx=ctx[-1])
        net(X)
        net.load_weights(args.params, fine_tune=num_classes != 80)
    else:
        print("params {} load error!".format(args.params))
        exit()
@Fermes
Copy link
Owner

Fermes commented Aug 2, 2018

Just delete this part of codes to disable finetune. If you want to change network structure, edit darknet.py. And actualize this idea in http://pjreddie.com/darknet/ is more recommended. ^.^

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants