Skip to content

Retrain Pretrained Model #53

Open
Open
@Wazz37

Description

I have my own dataset which I want to retrain using the provided pretrianed model. I set the path of pretrained model downloaded from the Reame.md but I am getitng this error -

[LINE:118] INFO [2023-09-26 11:31:35,803] Database for no_gesture not found [LINE:118] INFO [2023-09-26 11:31:35,958] Database for no_gesture not found [LINE:83] INFO [2023-09-26 11:31:35,966] Current device: cpu Traceback (most recent call last): File "/home/hitech/anaconda3/envs/gestures/lib/python3.9/runpy.py", line 197, in _run_module_as_main return _run_code(code, main_globals, None, File "/home/hitech/anaconda3/envs/gestures/lib/python3.9/runpy.py", line 87, in _run_code exec(code, run_globals) File "/home/hitech/Programming/hagrid/detector/run.py", line 104, in <module> _run_train(args.path_to_config) File "/home/hitech/Programming/hagrid/detector/run.py", line 84, in _run_train TrainDetector.train(model, conf, train_dataset, test_dataset) File "/home/hitech/Programming/hagrid/detector/train.py", line 125, in train params = [p for p in model.parameters() if p.requires_grad] AttributeError: 'NoneType' object has no attribute 'parameters'

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions