search

KaLiMaLi555 / model_extraction_interiit

1 stars 0 forks source link

Added logits caching script for TF MoViNet #10

Closed IshanManchanda closed 2 years ago

IshanManchanda commented 2 years ago

Closes #3

shreyas-kowshik commented 2 years ago

@IshanManchanda This uses a pytorch dataloader I suppose. So I suppose that would not work with passing that through a Keras model right? Or did you happen to test and it's working?

IshanManchanda commented 2 years ago

@IshanManchanda This uses a pytorch dataloader I suppose. So I suppose that would not work with passing that through a Keras model right? Or did you happen to test and it's working?

The torch dataloader is used to instantiate a TF Dataset. It's been tested and even used to generate the K600 logits for noise and the val set on colab. Refer lines 71 and 72: https://github.com/KaLiMaLi555/model_extraction_interiit/pull/10/files#diff-370da2879999cd0d9a7cd01a45252e39245eb8e8c15a2d53f674bb6cfec545ccR71

shreyas-kowshik commented 2 years ago

Sounds good! Merging