We use n_samples often to debug for memory size on the GPU. However, peak memory is determined by the biggest samples in our data. Before we have seen all of them takes 1 whole epoch. To shorten this process, n_samples now returns the n largest data points.
We use
n_samplesoften to debug for memory size on the GPU. However, peak memory is determined by the biggest samples in our data. Before we have seen all of them takes 1 whole epoch. To shorten this process, n_samples now returns thenlargest data points.