Skip to content

train my own model #1310

@wmkai

Description

@wmkai

I want to train my own model from scratch, but the log shows that

Training with a single process on 1 GPUs.
Traceback (most recent call last):
  File "/home/tiger/code/pytorch-image-models/train.py", line 866, in <module>
    main()
  File "/home/tiger/code/pytorch-image-models/train.py", line 397, in main
    model = create_model(
  File "/home/tiger/code/pytorch-image-models/timm/models/factory.py", line 71, in create_model
    model = create_fn(pretrained=pretrained, pretrained_cfg=pretrained_cfg, **kwargs)
  File "/home/tiger/code/pytorch-image-models/timm/models/topformer.py", line 662, in topformer_base
    return _create_topformer('topformer_base', pretrained, **model_args)
  File "/home/tiger/code/pytorch-image-models/timm/models/topformer.py", line 575, in _create_topformer
    return build_model_with_cfg(Topformer, variant, pretrained, **kwargs)
  File "/home/tiger/code/pytorch-image-models/timm/models/helpers.py", line 511, in build_model_with_cfg
    pretrained_cfg = resolve_pretrained_cfg(variant, pretrained_cfg=pretrained_cfg)
  File "/home/tiger/code/pytorch-image-models/timm/models/helpers.py", line 469, in resolve_pretrained_cfg
    assert pretrained_cfg
AssertionError
(torch-env) 

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions