十字绣网络的多任务学习
该项目是TensorFlow实现的一种多任务学习方法,该方法在论文《。
争论
--lr,学习率
--n_epoch,纪元数
--n_batch_size,最小批处理大小
--reg_lambda,L2正则化lambda
--keep_prob,退出保持概率
--cross_stitch_enabled,是否使用十字绣
数据集
时尚MNIST
是Zalando文章图片的数据集,由6万个示例的训练集和10,000个示例的测试集组成。 每个示例都是一个28x28灰度图像,与来自10个类别的标签相关联:
标签
描述
标签
描述
0
T恤/上衣
5
凉鞋
1个
裤子
6
衬衫
2个
套衫
7
运动鞋
3
裙子
8
包
4
外套
9
脚踝靴
为了进行多任务学习,我为每个图像创建了另一个标签,该标签基于原始标签:
标签
原始标签
描述
0
5、7、9
2021-09-17 09:55:00
382KB
Python
1