Closed petebankhead closed 1 year ago
NDArray
DjlTools.getXXX()
DjlDnnModel
DjlTools.get/setOverrideDevice()
Device.fromName('mps')
NDArray
to more data typesDjlTools.getXXX()
methods to get ints, floats, doubles, longs and booleansDjlDnnModel
if shape doesn't match NDLayoutDjlTools.get/setOverrideDevice()
methods to override DJL's default device selectionDevice.fromName('mps')
on Apple Silicon (which sometimes works, sometimes doesn't...)