code: def main(config: DictConfig) -> None:
      corpus = UITABSAHotel(
          training="aspect"
      )  # predicting aspect or polarity
      batch_size = 24
      max_sequence_len = 100
      num_labels = corpus.num_labels
      BERT_MODEL_NAME = "vinai/phobert-base"
  
      tokenizer = AutoTokenizer.from_pretrained(
          BERT_MODEL_NAME, use_fast=False
      )
      train_dataset = UITABSADataset(
          data=corpus.train,
          max_sequence_len=max_sequence_len,
          num_labels=num_labels,
          tokenizer=tokenizer,
      )
      val_dataset = UITABSADataset(
          data=corpus.dev,
          max_sequence_len=max_sequence_len,
          num_labels=num_labels,
          tokenizer=tokenizer,
      )
      test_dataset = UITABSADataset(
          data=corpus.test,
          max_sequence_len=max_sequence_len,
          num_labels=num_labels,
          tokenizer=tokenizer,
      )
  
      n_warmup_steps = 20
      n_training_steps = 100
  
      model = BertForMultilabelClassification(
          bert_model=BERT_MODEL_NAME,
          n_classes=num_labels,
          n_warmup_steps=n_warmup_steps,
          n_training_steps=n_training_steps,
      )
  
      data_module = UITABSADataModule(
          train_dataset, val_dataset, test_dataset, batch_size
      )
  
      # logger = WandbLogger(project="debug-phobert-sentiment")
      trainer = pl.Trainer(
          max_epochs=config.trainer.epoch,
          accelerator="cpu",
          enable_progress_bar=True,
          # logger=logger
      )
      trainer.fit(model, data_module)
      trainer.test()
  
imports: import torch
  import hydra
  import torch.nn
  import pytorch_lightning
  from omegaconf import DictConfig
  from torchmetrics import F1Score
  from torch.utils.data import Dataset, DataLoader
  from transformers import AutoTokenizer, AdamW, RobertaModel, BertPreTrainedModel, RobertaConfig
  from underthesea import word_tokenize
  from underthesea.datasets.uit_absa_hotel.uit_absa_hotel import UITABSAHotel
label_desc: Use image node_id 1 for calling a global function with example usage: main(config) without return types