Minor changes; add train_timestep_fraction

2023-06-27 22:17:32 -07:00
parent bae3f43f5f
commit 28d2d8c40e
5 changed files with 50 additions and 26 deletions
--- a/ddpo_pytorch/rewards.py
+++ b/ddpo_pytorch/rewards.py
@@ -35,8 +35,6 @@ def aesthetic_score():
    scorer = AestheticScorer().cuda()

    def _fn(images, prompts, metadata):
-        if not isinstance(images, torch.Tensor):
-            images = torch.as_tensor(images)
        scores = scorer(images)
        return scores, {}