posted an update

# Usage: def get_classification_token(batch) -> torch.Tensor: """Process embeddings and return classification token.""" model = ProteinInteractionTransformer() return model(batch, num_embeddings=len(batch), embedding_dimensions=batch[0].size(0))

Log in or sign up for Devpost to join the conversation.