train_config.json

{
    "seed": 0,
    "run_name": "try1",
    "saved_dir" : "./results",
    "train_dataset" : {"name" : "CustomDataset",
                       "args" : {"data_path_main_train" : "./data/movielens/main_data_train_ml.csv",
                                 "data_path_main_test" : "./data/movielens/main_data_test_ml.npy",
                                 "data_path_aux_user" : "./data/movielens/ml_users2.csv",
                                 "data_path_aux_item" : "./data/movielens/ml_movies2.csv",
                                 "num_ng" : 4,
                                 "is_training" : true}},
    "val_dataset" : {"name" : "CustomDataset",
                     "args" : {"data_path_main_train" : "./data/movielens/main_data_train_ml.csv",
                               "data_path_main_test" : "./data/movielens/main_data_test_ml.npy",
                               "data_path_aux_user" : "./data/movielens/ml_users2.csv",
                               "data_path_aux_item" : "./data/movielens/ml_movies2.csv",
                               "num_ng" : 0,
                               "is_training" : false}},
    "train_dataloader": {"args" : {"batch_size" : 64,
                                   "num_workers" : 0,
                                   "shuffle" : true,
                                   "drop_last" : true}},
    "val_dataloader" : {"args" : {"batch_size" : 100,
                                  "num_workers" : 0,
                                  "shuffle" : false,
                                  "drop_last" : true}},
    "num_epochs": 100,
    "model": {"name" : "ViT",
              "args" : {"user_num" : 6040,
                        "item_num" : 3952,
                        "emb_size" : 8,
                        "factor_num" : 16,
                        "patch_size" : 2,
                        "depth" : 2,
                        "aux_depth" : 2,
                        "user_out" : 21,
                        "item_out" : 18}},
    "resume_from": false,
    "resume_mode": "all",
    "checkpoint_path": "",
    "criterion": {"name" : "CustomLoss",
                  "args": {"lambda_main" : null,
                           "lambda_user" : null,
                           "lambda_item" : null}},
    "optimizer": {"name" : "Adam",
                  "args" : {"lr" : 0.01}},
    "scheduler":{"name":"ReduceLROnPlateau",
        "args":{"mode" : "max",
                "factor" : 0.9,
                "patience" : 3}},
    "top_k" : 10,
    "val_every" : 1,
    "save_mode": "hr",
    "num_to_remain": 3,
    "fp16": false
}