Merge pull request #48 from RMeli/issue-47

Fix random translation
RMeli · Aug 18, 2022 · 4b83656 · 4b83656
2 parents 34650b8 + 754bffc
commit 4b83656
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
@@ -44,7 +44,7 @@ jobs:
 
  # TODO: Install package and run tests from TestPyPI
 
- - name: Publish on TestPyPI
+ - name: Publish on PyPI
  if: github.event_name == 'release' && github.event.action == 'published'
  uses: pypa/[email protected]
  with:

diff --git a/gninatorch/dataloaders.py b/gninatorch/dataloaders.py
@@ -128,7 +128,7 @@ def __next__(self):
  self.grid_maker.forward(
  batch,
  grids,
- random_translation=self.random_rotation,
+ random_translation=self.random_translation,
  random_rotation=self.random_rotation,
  )