Skip to content

Commit 7c4ce39

Browse files
authored
fix: Fix examples and dragan epoch checkpoint (#106)
* fix examples and dragan train checkpoint * gan_example fix 🚀
1 parent 3c5f1ee commit 7c4ce39

4 files changed

Lines changed: 155 additions & 452 deletions

File tree

examples/regular/cgan_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
#Before training the GAN do not forget to apply the required data transformations
2121
#To ease here we've applied a PowerTransformation
22-
data = transformations(data)
22+
_, data, _ = transformations(data)
2323

2424
#For the purpose of this example we will only synthesize the minority class
2525
train_data = data.loc[ data['Class']==1 ].copy()

0 commit comments

Comments
 (0)