Readd mps support

models/RNN_no_attention_unidirectional/RNN_no_attention_unidirectional.py

@@ -97,8 +97,7 @@ class Seq2Seq(nn.Module):
 # setup environment
 source_data_path = str(project_root / "data/tokenizer-data/news-commentary-v11.en")
 target_data_path = str(project_root / "data/tokenizer-data/news-commentary-v11.de")
-#device = utils.pytorch.get_available_device()
-device = torch.device("cpu")
+device = utils.pytorch.get_available_device()
 date_time_now = datetime.now().strftime("%m%d%Y_%H%M")
 model_output_path = str(work_dir / f"./checkpoints/{date_time_now}_RNN_no_attention_unidirectional.pt")
 

models/RNN_with_attention_bidirectional/RNN_with_attention_bidirectional.py

@@ -133,8 +133,7 @@ class Seq2Seq(nn.Module):
 # setup environment
 source_data_path = str(project_root / "data/tokenizer-data/news-commentary-v11.en")
 target_data_path = str(project_root / "data/tokenizer-data/news-commentary-v11.de")
-#device = utils.pytorch.get_available_device()
-device = torch.device("cpu")
+device = utils.pytorch.get_available_device()
 date_time_now = datetime.now().strftime("%m%d%Y_%H%M")
 model_output_path = str(work_dir / f"./checkpoints/{date_time_now}_RNN_attention_bidirectional.pt")