search

hjsuh94 / score_po

Score-Guided Planning
10 stars 0 forks source link

Change several class to torch.nn.Module: #32

Closed hongkai-dai closed 1 year ago

hongkai-dai commented 1 year ago
  1. Cost
  2. DataDistanceEstimator
  3. DynamicalSystem
  4. ScoreEstimator

Specifically, the previously saved ScoreEstimator checkpoint cannot be used any more, because the key in the stored checkpoint file is changed.

This change is Reviewable