Skip to content

Instantly share code, notes, and snippets.

@adamoudad

adamoudad/initialize_training_imdb_pytorch.py

Created March 20, 2021 21:32
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save adamoudad/bbaf0b3a5b5d609c5db06f1a7e86ddb9 to your computer and use it in GitHub Desktop.
Save adamoudad/bbaf0b3a5b5d609c5db06f1a7e86ddb9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
initialize_training_imdb_pytorch.py
model = BiLSTM(input_dim, embedding_dim, hidden_dim)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
criterion = nn.BCELoss().to(device)
optimizer = torch.optim.Adam(model.parameters())
model.to(device)
batch_history = {
"loss": [],
"accuracy": []
}
epoch_history = {
"loss": [],
"accuracy": [],
"val_loss": [],
"val_accuracy": [],
}
@0707jim
Copy link

0707jim commented Mar 20, 2021

initialize_training_imdb_pytorch.py

@0707jim
Copy link

0707jim commented Mar 20, 2021

initialize_training_imdb_pytorch.py

@0707jim
Copy link

0707jim commented Mar 20, 2021

initialize_training_imdb_pytorch.py

@0707jim
Copy link

0707jim commented Mar 20, 2021

#initialize_training_imdb_pytorch.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment