Searched defs:model_cls (Results 1 – 1 of 1) sorted by relevance
130 def get_sequence_length(model_cls, model_name): argument172 model_cls, model, model_name, bs, device, include_loss_args=False argument