hutaotao1
hutaotao1
补充一下问题：train.py里面指定 --device default='gpu:0'时会报错，这个方法行不通。是需要重新配置CUDA到GPU1上吗？
您发给我的邮件我已经收到！
> Remove [0] in train.py Tha's it! Thank you a lot!!!!
您发给我的邮件我已经收到！
您发给我的邮件我已经收到！
胡涛涛 ***@***.*** 你好！ 之前的回答被close了，我也忘记详细的代码了。 应该是paddleseg文件夹里面有个core的文件夹， 然后里面的train.py，有一句代码是loss.data.....什么的，后面有个numpy()[0], 需要把这里的[0]给删了，就可以了。 这个报错是paddleseg的源代码问题，各个版本之间不兼容。你可以这样修改看看，我只是依稀记得是这样搞定的，如果不对的话，你发详细的代码过来我可以帮你看看。 Original Email Sender:"X-MingYang"< ***@***.*** >; Sent Time:2024/3/16 19:34 To:"PaddlePaddle/PaddleSeg"< ***@***.*** >; Cc recipient:"hutaotao1"< ***@***.*** >;"Comment"< ***@***.*** >; Subject:Re: [PaddlePaddle/PaddleSeg] 出现IndexError: too many...
不客气。 胡涛涛 ***@***.*** Original Email Sender:"X-MingYang"< ***@***.*** >; Sent Time:2024/3/16 21:19 To:"PaddlePaddle/PaddleSeg"< ***@***.*** >; Cc recipient:"hutaotao1"< ***@***.*** >;"Comment"< ***@***.*** >; Subject:Re: [PaddlePaddle/PaddleSeg] 出现IndexError: too many indices for array: array is...
这个除了修改这里还要修改： 1./home/aistudio/PaddleSeg/paddleseg/core/train.py 文件内'avg_loss += loss.numpy()[0]'改成‘avg_loss += loss.numpy()’2.‘avg_loss_list = [l[0] / log_iters for l in avg_loss_list]’也是要改成‘avg_loss_list = [l / log_iters for l in avg_loss_list]’。 3.metrics.py文件里的‘pred_area.append(paddle.sum(paddle.cast(pred_i, "int32")))’也要改成‘pred_area.append(paddle.sum(paddle.cast(pred_i, "int32")).unsqueeze(0))’ 这样才能正常运行。 ———————————————— ...
不是，我记得就是要这样改的。 胡涛涛 ***@***.*** Original Email Sender:"X-MingYang"< ***@***.*** >; Sent Time:2024/3/16 22:39 To:"PaddlePaddle/PaddleSeg"< ***@***.*** >; Cc recipient:"hutaotao1"< ***@***.*** >;"Comment"< ***@***.*** >; Subject:Re: [PaddlePaddle/PaddleSeg] 出现IndexError: too many indices for array: array is...