validate() takes 2 positional arguments but 3 were given

在运行程序时,定位到以下语句出现错误:validate(test_loader, model, n_labels)

报错信息为:validate() takes 2 positional arguments but 3 were given

if val_f1 >= best_f1:
    best_f1 = val_f1
    test_output_scores, test_f1 = validate(test_loader, model, n_labels)
    all_test_f1.append(test_f1)

函数validate设置了两个形式参数,在调用的时候给了三个参数,所以就会出现报错;因为主程序中validate在下面还这样调用,所以只能去修改validate()

# def validate(val_loader, model):
def validate(val_loader, model, n_class):
    model.eval()

定位到validate(),发现确实如此,于是多加个参数,问题解决。

 

你可能感兴趣的:(自然语言处理,pytorch)