search

jstmn / ikflow

Open source implementation to the paper "IKFlow: Generating Diverse Inverse Kinematics Solutions"
https://sites.google.com/view/ikflow/home
Other
35 stars 5 forks source link

Set devices to auto to enable auto identification from the system #9

Closed xyyeh closed 2 months ago

xyyeh commented 2 months ago

Replace GPU_IDX definitions in all train scripts. Use device="auto" to enable lightning to auto specify max count from the system

xyyeh commented 2 months ago

Replace GPU_IDX definitions in all train scripts. Use device="auto" to enable lightning to auto specify max count from the system

GPU_IDX was note defined previously but used in all train scripts