我正在用火把开发CNN。我的模型在没有增强的训练和测试集上都有很好的准确性,但是我想学习增强,所以我使用了火炬视觉变换来增强,在应用了增强模型之后,开始做最坏的事情,损失也没有减少。所以我试着调试并观察到增强的图像看起来是扭曲的/出乎意料的,有人能帮我解决这个问题吗?class traindataset(Dataset): '''
data: data is a pandas dataframe g
当我尝试重塑每个权重矩阵以达到1维时,问题就出现了(这对于我稍后使用梯度进行计算很有用)。在这一点上,Tensorflow输出一个None列表(这意味着没有路径从损失到这些张量,而应该有,因为它们是模型参数重塑)。如果我删除整形线all_tensors = [tf.reshape(x, [-1]) for x in all_tensors],代码将正常工作,并返回4个张量,每个参数包含梯度wrt。我非常确定,重塑不会破坏图中的任何依赖关系,否则它根本不能在任何网络中使用。