ADMM-神经网络
训练没有梯度的神经网络:一种ADMM方法(具有急切执行力和Matplotlib的Matlab,Tensorflow1.6-Python2.7 / 3.5)注意:您不能将此代码用于任何分配或任何盈利产品。
该代码是根据以下论文实施的; 但是,我使用均方误差损失而不是二进制铰链损失
泰勒,加文等。 “训练没有梯度的神经网络:可扩展的admm方法。” 机器学习国际会议。 2016年。
仅支持GPU?
我只为GPU版本实现并在MNIST数据上进行了测试。
怎么跑?
运行main.m / main.py
2021-11-15 17:09:46
29.03MB
Python
1