Skip to content

Commit 9b73802

Browse files
committed
lsturcon/supconloss fix andreeaiana#3
1 parent 3b183b1 commit 9b73802

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

configs/experiment/lsturcon_rumind_pretrainedemb_supconloss_bertsent.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ data:
2222

2323
model:
2424
loss: "sup_con_loss"
25-
num_categ_classes: 26
25+
num_categ_classes: 28
2626
use_plm: False
2727
pretrained_embeddings_path: ${paths.data_dir}MINDcombined_train/transformed_word_embeddings.npy
2828
text_embed_dim: 300
@@ -31,7 +31,7 @@ model:
3131
window_size: 3
3232
query_dim: 200
3333
dropout_probability: 0.2
34-
num_users: 134
34+
num_users: 133
3535
long_short_term_method: "con" # choose between "ini" and "con"
3636

3737
callbacks:

0 commit comments

Comments
 (0)