Make sure every module is compatible with the to device conversion

Yann-CV / vintage-models

Pytorch personal implementation of the most famous neural network models

GNU General Public License v3.0

0 stars 0 forks source link

Closed Yann-CV closed 4 months ago

Yann-CV commented 4 months ago

this is also fixing some inconsistencies regarding trainablesumpool

Yann-CV commented 4 months ago

This stack of pull requests is managed by Graphite. Learn more about stacking.

Join @Yann-CV and the rest of your teammates on Graphite