-
PyTorch中实现的CLEVR关系网络:relation-networks-pytorch
资源介绍
关系网络-火炬
在PyTorch中实现的用于CLEVR的关系网络( )
要求:
Python 3.6
火炬
火炬视觉
枕头
恩特克
tqdm
训练:
从下载并提取CLEVR v1.0数据集
预处理问题数据
python preprocess.py [CLEVR directory]
运行train.py
python train.py [CLEVR directory]
重现状态
最后转载了此,感谢 ! ( )重现结果的关键配置是大批量(640),每20个周期将学习率提高到2倍,直到达到最大学习率,然后反转问题,是,以相反的顺序将疑问词输入LSTM。
精度图