text-graph-generalization

Code for our paper Compositional Generalization with Grounded Language Models
git clone https://esimon.eu/repos/text-graph-generalization.git
Log | Files | Refs | README | LICENSE

device.py (71B)

      1 import torch
      2 device = 'cuda:0' if torch.cuda.is_available() else 'cpu'