lonePatient / Bert-Multi-Label-Text-Classification

This repo contains a PyTorch implementation of a pretrained BERT model for multi-label text classification.

Why restoring checkpoint is run on CPU not GPU?

Sajjadahmed668 opened this issue 5 years ago · comments

Sajjad Ahmed commented 5 years ago

Why restoring checkpoint is run on CPU not GPU?

千帆竞发 commented 5 years ago

Have you solved this problem? I always use CPU when running programs, which is very strange.