grasp_diffusion icon indicating copy to clipboard operation
grasp_diffusion copied to clipboard

improve batch-wise grasp generation in sampling scripts

Open kuldeepbrd1 opened this issue 1 year ago • 1 comments

To resolve issue #4

  • added consistent use of batch_size and device variables
  • added sampling loop to support desired n_grasps
  • added device and batch_size as cli args
  • Auto-sorted imports and format changes from black

kuldeepbrd1 avatar Nov 01 '22 20:11 kuldeepbrd1