Tsinghua-MARS-Lab / StateTransformer

210 stars 18 forks source link

fix invalid ending point bug #63

Closed larksq closed 1 year ago

larksq commented 1 year ago
  1. fix a bug causing loss to generate nan and explode the training! due to wrong invalid goal point checking
  2. update end point invalid checking during dataset generation
  3. add multi-processing into data generation