1训练时注意加载数据文件地址
2.训练代码中loss[0]应该改为loss
3.训练过程中,出现ValueError: some of the strides of a given numpy array are negative. This is currently not supported错误
应该将代码inputs = inputs.type(torch.FloatTensor) labels = labels.type(torch.FloatTensor)
改为inputs = np.ascontiguousarray(inputs, dtype = np.float32) inputs = Variable(torch.FloatTensor(inputs))
U_2net网络代码训练问题记录
最新推荐文章于 2024-08-08 08:24:49 发布