def build_model(config):
vocab = Vocab(config['vocab'])
device = config['device']
model = VietOCR(len(vocab), config['backbone'], config['cnn'],
config['transformer'], config['seq_modeling'])
model = model.to(device)
return model, vocab
def build_model(config):
vocab = Vocab(config['vocab'])
device = config['device']
model = VietOCR(len(vocab),
ss=config['cnn']['pooling_stride_size'], ks=config['cnn']['pooling_kernel_size'],
**config['transformer'])
model = model.to(device)
return model, vocab