search

cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
http://contextualized.ml/
GNU General Public License v3.0
65 stars 9 forks source link

Batched MSE #160

Open cnellington opened 1 year ago

cnellington commented 1 year ago

measure_mses in easy networks needs batched network prediction. Loading all networks into memory before measuring MSEs can sometimes cause OOM. https://github.com/cnellington/Contextualized/blob/30cb09c0261899584c61eb93997f1ca8a624ca3e/contextualized/easy/ContextualizedNetworks.py#L75