Skip to content

fix bug after train on evaluation#38

Open
wangqiang1588 wants to merge 1 commit intoMhLiao:masterfrom
wangqiang1588:master
Open

fix bug after train on evaluation#38
wangqiang1588 wants to merge 1 commit intoMhLiao:masterfrom
wangqiang1588:master

Conversation

@wangqiang1588
Copy link
Copy Markdown
Contributor

2019-12-02 11:07:34,096 maskrcnn_benchmark.inference INFO: Start evaluation on 233 images

0it [00:01, ?it/s]
Traceback (most recent call last):
File "tools/train_net.py", line 174, in
main()
File "tools/train_net.py", line 170, in main
test(cfg, model, args.distributed)
File "tools/train_net.py", line 106, in test
output_folder=output_folder,
File "/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 371, in inference
predictions = compute_on_dataset(model, data_loader, device)
File "
/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 32, in compute_on_dataset
output = [o.to(cpu_device) for o in output]
File "~/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 32, in
output = [o.to(cpu_device) for o in output]
AttributeError: 'list' object has no attribute 'to'

2019-12-02 11:07:34,096 maskrcnn_benchmark.inference INFO: Start evaluation on 233 images

0it [00:01, ?it/s]
Traceback (most recent call last):
  File "tools/train_net.py", line 174, in <module>
    main()
  File "tools/train_net.py", line 170, in main
    test(cfg, model, args.distributed)
  File "tools/train_net.py", line 106, in test
    output_folder=output_folder,
  File "~/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 371, in inference
    predictions = compute_on_dataset(model, data_loader, device)
  File "~/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 32, in compute_on_dataset
    output = [o.to(cpu_device) for o in output]
  File "~/MaskTextSpotter/maskrcnn_benchmark/engine/inference.py", line 32, in <listcomp>
    output = [o.to(cpu_device) for o in output]
AttributeError: 'list' object has no attribute 'to'
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

Successfully merging this pull request may close these issues.

1 participant