Spaces:
Running
on
T4
Running
on
T4
Hugo Flores
commited on
Commit
•
79bcce6
1
Parent(s):
9439b64
fix random seeds for train!
Browse files- scripts/exp/train.py +2 -1
scripts/exp/train.py
CHANGED
@@ -264,6 +264,7 @@ def train(
|
|
264 |
):
|
265 |
assert codec_ckpt is not None, "codec_ckpt is required"
|
266 |
|
|
|
267 |
at.util.seed(seed)
|
268 |
writer = None
|
269 |
|
@@ -282,7 +283,7 @@ def train(
|
|
282 |
sample_rate = codec.sample_rate
|
283 |
|
284 |
# a better rng for sampling from our schedule
|
285 |
-
rng = torch.quasirandom.SobolEngine(1, scramble=True)
|
286 |
|
287 |
# log a model summary w/ num params
|
288 |
if accel.local_rank == 0:
|
|
|
264 |
):
|
265 |
assert codec_ckpt is not None, "codec_ckpt is required"
|
266 |
|
267 |
+
seed = seed + accel.local_rank
|
268 |
at.util.seed(seed)
|
269 |
writer = None
|
270 |
|
|
|
283 |
sample_rate = codec.sample_rate
|
284 |
|
285 |
# a better rng for sampling from our schedule
|
286 |
+
rng = torch.quasirandom.SobolEngine(1, scramble=True, seed=seed)
|
287 |
|
288 |
# log a model summary w/ num params
|
289 |
if accel.local_rank == 0:
|