Skip to content

Commit 61c08ca

Browse files
Merge pull request #125 from AI-Hypercomputer/cloudchrischan-patch-1
Update hstu.py
2 parents 847628b + 42091f1 commit 61c08ca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recml/layers/keras/hstu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def default_bucketization(tensor: Tensor, bucket_size: float = 0.301):
3535
"""Default bucketization function for RelativeBucketedTimeAndPositionBasedBias.
3636
3737
From the original paper's code base:
38-
https://github.com/facebookresearch/generative-recommenders/blob/main/modeling/sequential/hstu.py#L79
38+
https://github.com/meta-recsys/generative-recommenders/blob/main/generative_recommenders/research/modeling/sequential/hstu.py#L87
3939
4040
Args:
4141
tensor: The tensor of timestamps to bucketize.

0 commit comments

Comments
 (0)