Machine Learning, Sentiment Analysis, Vertex AI, Programming, Bert.
--- CONSTANTS --- BERT_MODEL_NAME = 'small_bert/bert_en_uncased_L-2_H-128_A-2' if torch.cuda.is_available(): logging.info(f"GPU: {torch.cuda.get_device_name(0)} is available.") DEVICE = torch.device('cuda') else: logging.info("No GPU available. Training will run on CPU.") DEVICE =