Skip to main content

Clashluke's group workspace

Timestamps visible
2023-05-10 06:01:42
    main()
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/main.py", line 202, in main
2023-05-10 06:01:42
    wctx = step(dat)
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/main.py", line 112, in __call__
2023-05-10 06:01:42
    self.wctx = WhileTrainContext(self.step(wctx.serialize()))
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/main.py", line 56, in jitless_step
2023-05-10 06:01:42
    return loop(train_step, wctx.serialize(), steps, training.device_unroll)
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 172, in loop
2023-05-10 06:01:42
    return lax.scan(lambda *x: (fn(*x[:-1]), None), fn_input, None, steps, unroll=unroll)[0]
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 172, in <lambda>
2023-05-10 06:01:42
    return lax.scan(lambda *x: (fn(*x[:-1]), None), fn_input, None, steps, unroll=unroll)[0]
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/main.py", line 29, in train_step
2023-05-10 06:01:42
    scalars, grads = grad_fn(params, data_slice)
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/main.py", line 99, in compute
2023-05-10 06:01:42
    return body_ctx(ctx, src, tgt)
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 35, in _fn
2023-05-10 06:01:42
    return fn(ctx, *args, **kwargs)
2023-05-10 06:01:42
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/main.py", line 90, in body_ctx
2023-05-10 06:01:42
    (out, loss), _ = lax.scan(block(ctx), carry, (src, tgt, jnp.arange(ctx.dims.sequence)))
2023-05-10 06:01:42
jax._src.linear_util.StoreException: Store occupied with not-equal value