update package version (#253)

* fix: save best final model

* fix date time format

* refactor common code

* refactor common code

* refactor

* fix: dataloader size for multi gpu

* added BaseModule for freeze batch norm

* added BaseModule

* updated models with BaseModule

* removed avg training time

* removed avg training time

* fix unit test

* added batch norm config to trainer

* fix unit test engine

* update step trainer

* empty cuda cache

* fix: ddp training

* added mock unit tests (#225)

* added mock unit tests (#233)

* merge ddp fix to development (#235)

* Bump version for DDP fix release

* Development (#234)

* Add tagged commit trigger for deployment workflow (#214)

* Update training config file to comply with the changes in the trainer (#215)

* Updated training config file to comply with changes in the trainer

* Made changes to default base trainer config as well

* Revert "Add tagged commit trigger for deployment workflow (#214)" (#217)

This reverts commit 6386fcb.

* fix training logs (#219)

* fix training logs

* setup local rank in ddp

* fix step trainer logs

* fix: save best final model

* fix date time format

* refactor common code

* refactor common code

* refactor

* fix: dataloader size for multi gpu

* added BaseModule for freeze batch norm

* added BaseModule

* updated models with BaseModule

* removed avg training time

* removed avg training time

* fix unit test

* added batch norm config to trainer

* fix unit test engine

* update step trainer

* empty cuda cache

* fix: ddp training

* added mock unit tests (#225)

* Bump ujson from 1.35 to 5.2.0 in /docs (#226)

Bumps [ujson](https://github.com/ultrajson/ultrajson) from 1.35 to 5.2.0.
- [Release notes](https://github.com/ultrajson/ultrajson/releases)
- [Commits](ultrajson/ultrajson@v1.35...5.2.0)

---
updated-dependencies:
- dependency-name: ujson
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Loosened requirements (#228)

* Loosened requirements (#229)

* Bump cookiecutter from 1.7.3 to 2.1.1 in /docs (#230)

Bumps [cookiecutter](https://github.com/cookiecutter/cookiecutter) from 1.7.3 to 2.1.1.
- [Release notes](https://github.com/cookiecutter/cookiecutter/releases)
- [Changelog](https://github.com/cookiecutter/cookiecutter/blob/master/HISTORY.md)
- [Commits](cookiecutter/cookiecutter@1.7.3...2.1.1)

---
updated-dependencies:
- dependency-name: cookiecutter
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump arrow from 0.13.1 to 0.15.1 in /docs (#231)

Bumps [arrow](https://github.com/arrow-py/arrow) from 0.13.1 to 0.15.1.
- [Release notes](https://github.com/arrow-py/arrow/releases)
- [Changelog](https://github.com/arrow-py/arrow/blob/master/CHANGELOG.rst)
- [Commits](arrow-py/arrow@0.13.1...0.15.1)

---
updated-dependencies:
- dependency-name: arrow
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump arrow from 0.13.1 to 0.15.1 (#232)

Bumps [arrow](https://github.com/arrow-py/arrow) from 0.13.1 to 0.15.1.
- [Release notes](https://github.com/arrow-py/arrow/releases)
- [Changelog](https://github.com/arrow-py/arrow/blob/master/CHANGELOG.rst)
- [Commits](arrow-py/arrow@0.13.1...0.15.1)

---
updated-dependencies:
- dependency-name: arrow
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Version bump

* added mock unit tests (#233)

Co-authored-by: Neelay Shah <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

Co-authored-by: NeelayS <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* bugfix: unbounded cropsize

* fix ddp tensorboard logging

* Added Augmentation Operations and unit tests (#237)

* Added new augment ops and normalization

* Added uni test for augmentation

* Changed uni test for augmentation

* Augmentation changes

* Fixed issue

* Added translate func fix and documentation

* Made suggested changes and  unit tested

* Normalize param issue fix

* Duplicate removal and norm_param style change

* Fix docstrings

Co-authored-by: Prajnan Goswami <[email protected]>

* fix tensorboard logging multi-gpu

* fix start step

* update step trainer

* :fix unit test

* Pwcnet refactor (#242)

* PWCNet refactor

* update pwcnet config

* fix model outputs

* fix flow output in evaluation

* fix engine unit tests

* update multiscale loss

* remove total loss logging for step trainer

* console output formatting

* fix dependency

* fix dependency

* remove normalization operation from the model

* update docstring

* update trainer

* update flownet C model

* update flownet c config

* bugfix FlowNetC

* fix flownet_c

* rearrage transformation sequence

* added chromatic augmentations

* added chromatic augmentations

* added chromatic augmentations

* added new spatial transforms to augmentor

* bugfix cropsize

* toggle transformations

* bug fix augmentation

* Augmentations (#243)

* added default behavior

* bugfix: default augmentation behavior

* bugfix: chromatic aug

* fix noise transform

* update trainer

* update flownet_c upsampling

* update logging in trainer

* update ddp training

* bugfix distributed training

* fix min scale

* fix: flow not available in dataset for prediction

* sync main  (#251)

* Bump oauthlib from 3.1.1 to 3.2.1 in /docs (#245)

Bumps [oauthlib](https://github.com/oauthlib/oauthlib) from 3.1.1 to 3.2.1.
- [Release notes](https://github.com/oauthlib/oauthlib/releases)
- [Changelog](https://github.com/oauthlib/oauthlib/blob/master/CHANGELOG.rst)
- [Commits](oauthlib/oauthlib@v3.1.1...v3.2.1)

---
updated-dependencies:
- dependency-name: oauthlib
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump oauthlib from 3.1.1 to 3.2.1 (#244)

Bumps [oauthlib](https://github.com/oauthlib/oauthlib) from 3.1.1 to 3.2.1.
- [Release notes](https://github.com/oauthlib/oauthlib/releases)
- [Changelog](https://github.com/oauthlib/oauthlib/blob/master/CHANGELOG.rst)
- [Commits](oauthlib/oauthlib@v3.1.1...v3.2.1)

---
updated-dependencies:
- dependency-name: oauthlib
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump protobuf from 3.17.3 to 3.18.3 in /docs (#247)

Bumps [protobuf](https://github.com/protocolbuffers/protobuf) from 3.17.3 to 3.18.3.
- [Release notes](https://github.com/protocolbuffers/protobuf/releases)
- [Changelog](https://github.com/protocolbuffers/protobuf/blob/main/generate_changelog.py)
- [Commits](protocolbuffers/protobuf@v3.17.3...v3.18.3)

---
updated-dependencies:
- dependency-name: protobuf
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump protobuf from 3.17.3 to 3.18.3 (#246)

Bumps [protobuf](https://github.com/protocolbuffers/protobuf) from 3.17.3 to 3.18.3.
- [Release notes](https://github.com/protocolbuffers/protobuf/releases)
- [Changelog](https://github.com/protocolbuffers/protobuf/blob/main/generate_changelog.py)
- [Commits](protocolbuffers/protobuf@v3.17.3...v3.18.3)

---
updated-dependencies:
- dependency-name: protobuf
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Fix docs errors (#248)

* Removed version info from docs config

* Fixed autodoc errors

* Bump certifi from 2021.5.30 to 2022.12.7 in /docs (#250)

Bumps [certifi](https://github.com/certifi/python-certifi) from 2021.5.30 to 2022.12.7.
- [Release notes](https://github.com/certifi/python-certifi/releases)
- [Commits](certifi/python-certifi@2021.05.30...2022.12.07)

---
updated-dependencies:
- dependency-name: certifi
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

* Bump certifi from 2021.5.30 to 2022.12.7 (#249)

Bumps [certifi](https://github.com/certifi/python-certifi) from 2021.5.30 to 2022.12.7.
- [Release notes](https://github.com/certifi/python-certifi/releases)
- [Commits](certifi/python-certifi@2021.05.30...2022.12.07)

---
updated-dependencies:
- dependency-name: certifi
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Neelay Shah <[email protected]>

* removed chromatic aug

* set default augmentation params

* decouple flip transform

* update aug_params

* bugfix: random flip augmentation

* bugfix: color transformation

* fix engine unit test

* fix functional unit tests

* update Multiscale loss unit test

* fix pre-commit code-style

* fix augmentator color params

* remove duplicate transformations

* update docstring

* update docstrings

* update docstrings

* add Kubric dataset

* update READMe:

* add link to kubric .tfrecord convert script

* update model configs

* updated training configs

* updated training configs

* update package version

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: NeelayS <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Devroop Kar <[email protected]>
Co-authored-by: Neelay Shah <[email protected]>

setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.3
+current_version = 0.2.4
 commit = True
 tag = True
 

setup.py

@@ -8,7 +8,7 @@
 # Basic information
 NAME = "ezflow"
 DESCRIPTION = "A PyTorch library for optical flow estimation using neural networks"
-VERSION = "0.2.3"
+VERSION = "0.2.4"
 AUTHOR = "Neelay Shah"
 EMAIL = "[email protected]"
 LICENSE = "MIT"