Commit Graph
20 Commits
Author SHA1 Message Date
henryruhs ed0f6ae897 Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming 2025-03-11 14:43:10 +01:00
henryruhs 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params 2025-03-11 14:43:10 +01:00
henryruhs 085c493e18 Rename id_embedder to embedder, Tons of naming in training step, Introduce new IdentityLoss class 2025-03-11 14:43:09 +01:00
henryruhs 5a6e3393e2 Revert loss for the moment 2025-03-11 14:43:09 +01:00
harisreedharandhenryruhs 4d8433f54a changes 2025-03-11 14:43:09 +01:00
harisreedharandhenryruhs db44c91dd8 changes 2025-03-11 14:43:09 +01:00
henryruhs 39c0313202 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 3cf9711df0 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 39818a16df Fix CI 2025-03-11 14:43:09 +01:00
henryruhs b6b4f9f65b Improve lot of types, imports and names 2025-03-11 14:43:09 +01:00
henryruhs 611618e413 Make Embedding great again 2025-03-11 14:43:03 +01:00
henryruhs 989a81c751 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 9c15f584aa Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
harisreedharandhenryruhs fcb3390796 again again cleaning 2025-03-11 14:43:03 +01:00
harisreedharandhenryruhs 3b8b6442fc add infer and some cleaning 2025-03-11 14:43:03 +01:00
harisreedharandhenryruhs 008a221f55 cleaning 2025-03-11 14:43:03 +01:00
henryruhs 10826558b4 Fix indent to tabs 2025-03-11 14:43:03 +01:00
henryruhs 23ac63d55b Fix more typing 2025-03-11 14:43:03 +01:00
harisreedharandhenryruhs 9e1c71b498 clean 2025-03-11 14:43:03 +01:00
harisreedharandhenryruhs e6c2a64256 ugly training code 2025-03-11 14:43:02 +01:00