-
-
Notifications
You must be signed in to change notification settings - Fork 5.1k
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
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 workingSomething isn't working