diff options
Diffstat (limited to 'training/sampler.py')
-rw-r--r-- | training/sampler.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/sampler.py b/training/sampler.py index 8afe255..bdb3e90 100644 --- a/training/sampler.py +++ b/training/sampler.py | |||
@@ -129,7 +129,7 @@ class LossSecondMomentResampler(LossAwareSampler): | |||
129 | self._loss_history = np.zeros( | 129 | self._loss_history = np.zeros( |
130 | [self.num_timesteps, history_per_term], dtype=np.float64 | 130 | [self.num_timesteps, history_per_term], dtype=np.float64 |
131 | ) | 131 | ) |
132 | self._loss_counts = np.zeros([self.num_timesteps], dtype=np.int) | 132 | self._loss_counts = np.zeros([self.num_timesteps], dtype=int) |
133 | 133 | ||
134 | def weights(self): | 134 | def weights(self): |
135 | if not self._warmed_up(): | 135 | if not self._warmed_up(): |