real-stanford / diffusion_policy

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
https://diffusion-policy.cs.columbia.edu/
MIT License
1.09k stars 203 forks source link

fixed val_ratio dose not function as expected in dataset #56

Open LYouC opened 3 months ago

LYouC commented 3 months ago

When call get_validation_dataset,the fuction use the ~self.train_mask as mask. This would result in the dataset being entirely a test set except for the training set, meaning the expected val_ratio did not take effect. Of course, this could be due to a misunderstanding of the meaning of val_ratio on my part