Support to choose running training/generation on CPU

ybracke / transnormer

A lexical normalizer for historical spelling variants using a transformer architecture.

GNU General Public License v3.0

6 stars 1 forks source link

Open ybracke opened 7 months ago

ybracke commented 7 months ago

[ ] Change train_model.py and generate.py:
device = torch.device(CONFIGS["gpu"] if torch.cuda.is_available() else "cpu")
gpu_index = CONFIGS.get("gpu")
device = torch.device(gpu_index if gpu_index != None and torch.cuda.is_available() else "cpu")
[ ] Update README section for GPU