train pytorch model
PyTorch on Google Cloud: How To train PyTorch models on AI Platform
After creating the AI Platform Notebooks instance, you can start with your experiments. Let's look into the model specifics for the use case. For analyzing sentiments of the movie reviews in IMDB dataset, we will be fine-tuning a pre-trained BERT model from Hugging Face. Fine-tuning involves taking a model that has already been trained for a given task and then tweaking the model for another similar task. Specifically, the tweaking involves replicating all the layers in the pre-trained model including weights and parameters, except the output layer.