search

askforalfred / alfred

ALFRED - A Benchmark for Interpreting Grounded Instructions for Everyday Tasks
MIT License
352 stars 77 forks source link

Extracting Resnet Features for the Full Dataset #126

Closed agneet42 closed 1 year ago

agneet42 commented 1 year ago

I am trying to extract resnet features using the command python models/utils/extract_resnet.py --data data/full_2.1.0 --batch 32 --gpu --visual_model resnet18 --filename feat_conv.pt , for the full dataset I was hoping to understand how long (~estimate) this takes (on a GPU) and what are the total number of feat_conv.pt that I can expect at the end of succesfull execution?

MohitShridhar commented 1 year ago

@agneet42, sorry, I don't remember because it's been 3 years since I ran this script. But maybe time one/few batch(es), and then try to extrapolate how long the full dataset would take?

The total number of feat_conv.pt should be the size of this os.walk

agneet42 commented 1 year ago

Thanks @MohitShridhar , looks like I was able to generate all the features. Took close to 1 day and generated about 7080 samples!