pytorch-tensor-decompositions:卷积层的[1412.6553]和[1511.06530]张量分解方法...

PyTorch张量分解 这是卷积层的Tucker和CP分解的实现。 关于此的博客文章可以在找到。 它取决于来执行张量分解。 用法 根据微调VGG16训练模型： python main.py --train 。 应该有一个包含两个类别的数据集。 每个类别一个目录。 训练数据应进入一个名为“ train”的目录。 测试数据应进入名为“ test”的目录。 这可以通过标志--train_path和--test_path来控制。 我使用了 然后将模型保存到名为“模型”的文件中。 执行分解： python main.py --decompose这会将新模型保存到“ decomposed_model”中。 默认情况下，它使用Tucker分解。 要使用CP分解，请传递--cp。 微调分解后的模型： python main.py --fine_tune 参考 卷积层的CP分解在此处描述： ：