15 Commits
Author SHA1 Message Date
2e6394565a Next (#93)
* add sync batchnorm

* replace random.choice with hash

* fifty percent reduction

* fix discriminator input

* restore dataset.py

* Remove duplicates

* add discriminator_ratio to config

* fix onnx export bug: replace round() with int()

* Fix embedding naming

* Introduce ModelWithConfigCheckpoint callback (#86)

* Fix dist ini

* Style: Refactor typing and improve code clarity in training.py (#88)

* Add type casting for trainer params

* Add type casting for trainer params

* Add type casting for trainer params

* Remove inplace activations for torch.compile compatibility (#89)

* Fix README

* improvise with norm layers & weighted average

* add skip layer

* use gelu instead of leaky_relu

* cleanup

* cleanup

* Update dependencies

* Different defaults and enable validation

* Different defaults and enable validation

* Revert to higher batch size

* Just use copy over copy2

---------

Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
Co-authored-by: NeuroDonu <112660822+NeuroDonu@users.noreply.github.com>
Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
2025-09-06 19:12:29 +02:00
henryruhs 7490ead302 Update dependencies 2025-06-17 22:03:22 +02:00
henryruhs 03011200e4 Update dependencies 2025-04-23 21:35:45 +02:00
Henry Ruhs d2be8a386a Update dependencies 2025-03-24 09:09:29 +01:00
henryruhs dd0a2fe649 Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs 578b07a7f4 Add StatefulDataloader, Manual trigger scheduler 2025-03-11 14:43:10 +01:00
henryruhs 607c55ff1f Remove Numpy and CV2 to fully use Tensors 2025-03-11 14:43:10 +01:00
henryruhs 7d8cb146a4 Use latest numpy 2025-03-11 14:43:10 +01:00
harisreedhar d1bf54276d changes 2025-03-11 14:43:09 +01:00
harisreedhar b47c6b72ee changes 2025-03-11 14:43:09 +01:00
henryruhs d9e10a9f7c Use lightning over pytorch_lightning import, Configure tensorboard 2025-03-11 14:43:09 +01:00
henryruhs cd4b10c832 Fix imports and update dependencies 2025-03-11 14:43:09 +01:00
harisreedhar bb0e3b4a8a remove liveportrait and update requirements 2025-03-11 14:43:03 +01:00
Henry Ruhs bd4725b52f Update dependencies 2024-11-11 21:54:04 +01:00
henryruhs 1670eff879 Initial commit 2024-10-15 16:56:57 +02:00