Make torch impl of pnet similar to TF (mostly from Frank)

zhanglab-aim / cancer-net

Diagnosing cancers using deep learning.

GNU General Public License v2.0

2 stars 0 forks source link

Make torch impl of pnet similar to TF (mostly from Frank) #31

Closed ttesileanu closed 1 year ago

ttesileanu commented 1 year ago

Add changes from Frank to make torch implementation of PNet work as well as the tensorflow. Slightly reformat, make loss consistent between val and train, improve reproducibility.