Code for the ICML 2021 (long talk) paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"
1.36k
stars
209
forks
source link
How to set the config to create a stand_alone commandline demo ? #70
Closed
Ngheissari closed 1 year ago
Here is the function that loads the model and tokenizer. What would be the _config ??
using @ex.automain would result in the follwing warning:
WARNING - timm.models.helpers - No pretrained weights exist or were found for this model. Using random initialization.
Here is the function.
def prepare_model(_config): _config = copy.deepcopy(_config)