Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Patch ctgan batchsize1 #263

Merged
merged 11 commits into from
Apr 12, 2024
+1 −1
Merged

Patch ctgan batchsize1 #263

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
aeaaa34
reuse encoders
bvanbreugel Jan 26, 2024
adccef9
Merge branch 'vanderschaarlab:main' into main
bvanbreugel Feb 21, 2024
65b747e
ensure categorical encoder is trained on real and synthetic
bvanbreugel Feb 21, 2024
1272fb2
better transformer
bvanbreugel Feb 21, 2024
8d50c2c
remove unnecessary imports
bvanbreugel Feb 21, 2024
e86d2e2
better error message
bvanbreugel Feb 21, 2024
d21d008
compatbility with DDIM
bvanbreugel Feb 22, 2024
99d1905
Merge branch 'vanderschaarlab:main' into main
bvanbreugel Mar 4, 2024
4cc472f
Merge branch 'vanderschaarlab:main' into main
bvanbreugel Mar 19, 2024
fc68f0c
Fix CTGAN for batchsize=1
bvanbreugel Mar 19, 2024
a3877f2
Merge branch 'vanderschaarlab:main' into patch-ctgan-batchsize1
bvanbreugel Apr 11, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/synthcity/plugins/core/models/gan.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -682,7 +682,7 @@ def _loss_gradient_penalty(
interpolated = (
alpha * real_samples + ((1 - alpha) * fake_samples)
).requires_grad_(True)
d_interpolated = self.discriminator(interpolated).squeeze()
d_interpolated = self.discriminator(interpolated).squeeze(-1)
labels = torch.ones((len(interpolated),), device=self.device)

# Get gradient w.r.t. interpolates
Expand Down
Loading