Python标量转换一个Arrary

运行Pytorch模型时出现以下错误。请问如何转换为一个size-1数组?这是我下面的代码:我看过其他StackOverflow答案,但不知道该怎么做。抱歉,这个问题很基本。我希望有人能帮帮忙。非常感谢!

回溯(最近通话最近): 文件“ train3_Eval.py”,行177 print(“ MAE Total:%f”%MAE_for_all_epochs) TypeError:必须为实数,而不是列表

# main training loop
global_step = 0
best_test_error = 10000
MAE_for_all_epochs = []
MSE_for_all_epochs = []
for epoch in range(4):
    print("Epoch %d" % epoch)
    model.train()
    for images,paths in tqdm(loader_train):
        images = images.to(device)
        targets = torch.tensor([metadata['count'][os.path.split(path)[-1]] for path in paths]) # B
        targets = targets.float().to(device)

        # forward pass:
        output = model(images) # B x 1 x 9 x 9 (analogous to a heatmap)
        preds = output.sum(dim=[1,2,3]) # predicted cell counts (vector of length B)
        
        # backward pass:
        loss = torch.mean((preds - targets)**2)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()

        # logging:
        count_error = torch.abs(preds - targets).mean()
        writer.add_scalar('train_loss',loss.item(),global_step=global_step)
        writer.add_scalar('train_count_error',count_error.item(),global_step=global_step)

        print("Step %d,loss=%f,count error=%f" % (global_step,count_error.item()))

        global_step += 1
    
    mean_test_error = 0
    model.eval()
    for images,paths in tqdm(loader_test):
        images = images.to(device)
        targets = torch.tensor([metadata['count'][os.path.split(path)[-1]] for path in paths]) # B
        targets = targets.float().to(device)

        # forward pass:
        output = model(images) # B x 1 x 9 x 9 (analogous to a heatmap)
        preds = output.sum(dim=[1,3]) # predicted cell counts (vector of length B)

        # logging:
        #error = torch.abs(preds - targets).sum().data
        #squared_error = ((preds - targets)*(preds - targets)).sum().data

        #runnning_mae += error
        #runnning_mse += squared_error

        loss = torch.mean((preds - targets)**2)
        count_error = torch.abs(preds - targets).mean()
        mean_test_error += count_error
        writer.add_scalar('test_loss',global_step=global_step)
        writer.add_scalar('test_count_error',global_step=global_step)
        
        global_step += 1
        #store_MAE = 0
        #store_MSE = 0

        mean_test_error = mean_test_error.detach() / len(loader_test)
        #store_MAE += mean_test_error
        MAE_for_all_epochs.append(mean_test_error)

        mse = math.sqrt(loss / len(loader_test))
        #store_MSE +=mse
        MSE_for_all_epochs.append(mse)

        print("Test count error: %f" % mean_test_error)
        print("MSE: %f" % mse)

    if mean_test_error < best_test_error:
        best_test_error = mean_test_error
        torch.save({'state_dict':model.state_dict(),'optimizer_state_dict':optimizer.state_dict(),'globalStep':global_step,'train_paths':dataset_train.files,'test_paths':dataset_test.files},checkpoint_path)

print("MAE Total: %f" % MAE_for_all_epochs)
print("MSE Total: %f" % MSE_for_all_epochs)

model_mae= MAE_for_all_epochs / epoch
model_mse= MSE_for_all_epochs / epoch
print("Model MAE: %f" % model_mae)
print("Model MSE: %f" % model_mse)

相关文章

本文适合有 Python 基础的小伙伴进阶学习 作者:pwwang 一、...
前言 目前有个python应用需要在容器镜像内拉取git私有仓库的...
前言 当网络不稳定或应用页面加载有问题,可以设置等待,避免...
前言 map()、reduce()、filter()是python的三个高阶函数。所...
入门使用 # 示例代码 warframe = [&quot;saryn&quot...
前言 功能描述:批量重命名指定目录下的文件,文件名加前缀,...