#!/bin/bash
# Copyright (c) 2024 Bytedance Ltd. and/or its affiliate

# Use 8GPUs for batch_size = 8x16 = 128
cd code && python3 main.py \
--config_file IDNet/sasrec.yaml overall/ID.yaml \
--optim_args.learning_rate 1e-3 \
--loss nce \
--train_batch_size 16 \
--MAX_ITEM_LIST_LENGTH 50 \
--epochs 201 \
--dataset amazon_books \
--hidden_dropout_prob 0.5 \
--attn_dropout_prob 0.5 \
--num_negatives 512 \
--n_layers 4 \
--n_heads 4 \
--embedding_size 64 \
--inner_size 1 \
--show_progress True \
--update_interval 100 \
--optim_args.weight_decay 0.0 \
--stopping_step 10