search

feng-spinning / onn-simulation

A program using pytorch to simulate diffracive deep nerual network (D2NN), containing updates (relative to the original paper) of incoherent propagation, parallel computation, nonlinear activation, changing of propagating distance.
15 stars 4 forks source link

一个小问题 #3

Open Sulanya opened 2 months ago

Sulanya commented 2 months ago

你好，复现的过程中，我发现predict.py（以及predict_am.py）里的H和U1维度不匹配，以至于无法计算U2，请问这里是有什么问题我没有注意到吗 c075a9cc4e700fdceeac53f74524ead c075a9cc4e700fdceeac53f74524ead image

feng-spinning commented 2 months ago

你好，我更新完成train后忘记更新predict了。请参照最新predict文件夹中的格式进行预测。