Skip to main content

Clashluke's group workspace

Timestamps visible
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/reversible.py", line 47, in base
2023-05-15 16:49:14
    out = fn(new_ctx, inp, *inner_args)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 35, in _fn
2023-05-15 16:49:14
    return fn(ctx, *args, **kwargs)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/linear.py", line 83, in read
2023-05-15 16:49:14
    offset1, offset0, gates = input_fn(ctx, token, position, dense0, ctx.dims.features, ctx.dims.pointwise_features,
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 35, in _fn
2023-05-15 16:49:14
    return fn(ctx, *args, **kwargs)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/linear.py", line 64, in input_fn
2023-05-15 16:49:14
    token_embedding = input_embed(ctx, token, ctx.dims.vocab)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 35, in _fn
2023-05-15 16:49:14
    return fn(ctx, *args, **kwargs)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/model/linear.py", line 29, in input_embed
2023-05-15 16:49:14
    param = get_param(ctx, "inp_embd", [dim, ctx.dims.pointwise_features], std=1 / ctx.dims.pointwise_features)
2023-05-15 16:49:14
  File "/home/ubuntu/HomebrewNLP-Jax/src/backend.py", line 143, in get_param
2023-05-15 16:49:14
    raise ValueError(f"Couldn't find parameter {prefix_name}. {ctx.name_cache=}")
2023-05-15 16:49:14
ValueError: Couldn't find parameter /body_ctx:0/block:0/reversible:2/read:1/input_fn:2/input_embed:4/inp_embd. ctx.name_cache={'body_ctx': 0, 'batch_embedding': 2, 'block': 0, 'reversible': 2, 'read': 1, 'input_fn': 2, 'input_embed': 4, 'scale_norm_act_linear': 6, 'pos_and_scale': 1, 'scale_norm_act': 3, 'linear': 0, '_output': 1, 'write': 0}