You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
initial_user_embedding = nn.Parameter(F.normalize(torch.rand(args.embedding_dim).cuda(), dim=0)) # the initial user and item embeddings are learned during training as well
user_embeddings = initial_user_embedding.repeat(num_users, 1)
uers_embeddings.detach_() # Detachment is needed to prevent double propagation of gradient
every epoch ,after the first t-btach the uers_embeddings will detach_(), so each epoch the initial_user_embedding only train one time? total epoch times?
The text was updated successfully, but these errors were encountered:
initial_user_embedding = nn.Parameter(F.normalize(torch.rand(args.embedding_dim).cuda(), dim=0)) # the initial user and item embeddings are learned during training as well
user_embeddings = initial_user_embedding.repeat(num_users, 1)
uers_embeddings.detach_() # Detachment is needed to prevent double propagation of gradient
every epoch ,after the first t-btach the uers_embeddings will detach_(), so each epoch the initial_user_embedding only train one time? total epoch times?
The text was updated successfully, but these errors were encountered: