Commit History

fix map_location=torch.device('cpu')
36b277a

matteopilotto commited on

remove model.to('cpu')
16d81eb

matteopilotto commited on

load model checkpoint on cpu rather than on gpu
c37b2cf

matteopilotto commited on