Giters

THU-KEG / MAVEN-ERE

Source code and dataset for EMNLP 2022 paper "MAVEN-ERE: A Unified Large-scale Dataset for Event Coreference, Temporal, Causal, and Subevent Relation Extraction".

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to select specific GPUs

cppww opened this issue · comments

commented

How to select specific GPUs?

commented

In class EventEncoder, self.model = nn.DataParallel(self.model, device_ids=[X,X,X])

commented

In def to_cuda, set x = x.to(device) device = 'cuda:'+str(num) num = device_ids[0]