ulf1 / sentence-embedding-evaluation-german

Basically SentEval with German language downstream tasks
Apache License 2.0
0 stars 1 forks source link

The validation set seems to be np.array #18

Closed ulf1 closed 2 years ago

ulf1 commented 2 years ago

Error: TypeError: linear(): argument 'input' (position 1) must be Tensor, not numpy.ndarray

Look here https://github.com/ulf1/sentence-embedding-evaluation-german/blob/e5aa4e7c25b19cc44b6400e1e888ea1cb1a7e72d/sentence_embedding_evaluation_german/model.py#L212

When does it happen

params = {
    'datafolder': './datasets',
    'batch_size': 128, 
    'num_epochs': 100,
    'early_stopping': True,
    'split_ratio': 0.1,  # if early_stopping=True
    'patience': 10,  # if early_stopping=True
}