James Betker
|
ef8d5f88c1
|
Bring split gaussian nll out of split so it can be computed accurately with the rest of the nll component
|
2020-11-27 13:30:21 -07:00 |
|
James Betker
|
11d2b70bdd
|
Latent space playground work
|
2020-11-27 12:03:16 -07:00 |
|
James Betker
|
4ab49b0d69
|
RRDB disc work
|
2020-11-27 12:03:08 -07:00 |
|
James Betker
|
6de4dabb73
|
Remove srflow (modified version)
Starting from orig and re-working from there.
|
2020-11-27 12:02:06 -07:00 |
|
James Betker
|
5f5420ff4a
|
Update to srflow_latent_space_playground
|
2020-11-26 20:31:21 -07:00 |
|
James Betker
|
fd356580c0
|
Play with lambdas
|
2020-11-26 20:30:55 -07:00 |
|
James Betker
|
0c6d7971b9
|
Dataset documentation
|
2020-11-26 11:58:39 -07:00 |
|
James Betker
|
45a489110f
|
Fix datasets
|
2020-11-26 11:50:38 -07:00 |
|
James Betker
|
5edaf085e0
|
Adjustments to latent_space_playground
|
2020-11-25 15:52:36 -07:00 |
|
James Betker
|
205c9a5335
|
Learn how to functionally use srflow networks
|
2020-11-25 13:59:06 -07:00 |
|
James Betker
|
cb045121b3
|
Expose srflow rrdb
|
2020-11-24 13:20:20 -07:00 |
|
James Betker
|
f3c1fc1bcd
|
Dataset modifications
|
2020-11-24 13:20:12 -07:00 |
|
James Betker
|
f6098155cd
|
Mods to tecogan to allow use of embeddings as input
|
2020-11-24 09:24:02 -07:00 |
|
James Betker
|
b10bcf6436
|
Rework stylegan_for_sr to incorporate structure as an adain block
|
2020-11-23 11:31:11 -07:00 |
|
James Betker
|
519ba6f10c
|
Support 2x RRDB with 4x srflow
|
2020-11-21 14:46:15 -07:00 |
|
James Betker
|
cad92bada8
|
Report logp and logdet for srflow
|
2020-11-21 10:13:05 -07:00 |
|
James Betker
|
c37d3faa58
|
More adjustments to srflow_orig
|
2020-11-20 19:38:33 -07:00 |
|
James Betker
|
d51d12a41a
|
Adjustments to srflow to (maybe?) fix training
|
2020-11-20 14:44:24 -07:00 |
|
James Betker
|
6c8c35ac47
|
Support training RRDB encoder [srflow]
|
2020-11-20 10:03:06 -07:00 |
|
James Betker
|
5ccdbcefe3
|
srflow_orig integration
|
2020-11-19 23:47:24 -07:00 |
|
James Betker
|
f80acfcab6
|
Throw if dataset isn't going to work with force_multiple setting
|
2020-11-19 23:47:00 -07:00 |
|
James Betker
|
2b2d754d8e
|
Bring in an original SRFlow implementation for reference
|
2020-11-19 21:42:39 -07:00 |
|
James Betker
|
1e0d7be3ce
|
"Clean up" SRFlow
|
2020-11-19 21:42:24 -07:00 |
|
James Betker
|
d7877d0a36
|
Fixes to teco losses and translational losses
|
2020-11-19 11:35:05 -07:00 |
|
James Betker
|
b2a05465fc
|
Fix missing requirements
|
2020-11-18 10:16:39 -07:00 |
|
James Betker
|
5c10264538
|
Remove pyramid_disc hard dependencies
|
2020-11-17 18:34:11 -07:00 |
|
James Betker
|
6b679e2b51
|
Make grad_penalty available to classical discs
|
2020-11-17 18:31:40 -07:00 |
|
James Betker
|
8a19c9ae15
|
Add additive mode to rrdb
|
2020-11-16 20:45:09 -07:00 |
|
James Betker
|
2a507987df
|
Merge remote-tracking branch 'origin/gan_lab' into gan_lab
|
2020-11-15 16:16:30 -07:00 |
|
James Betker
|
931ed903c1
|
Allow combined additive loss
|
2020-11-15 16:16:18 -07:00 |
|
James Betker
|
4b68116977
|
import fix
|
2020-11-15 16:15:42 -07:00 |
|
James Betker
|
98eada1e4c
|
More circular dependency fixes + unet fixes
|
2020-11-15 11:53:35 -07:00 |
|
James Betker
|
e587d549f7
|
Fix circular imports
|
2020-11-15 11:32:35 -07:00 |
|
James Betker
|
99f0cfaab5
|
Rework stylegan2 divergence losses
Notably: include unet loss
|
2020-11-15 11:26:44 -07:00 |
|
James Betker
|
ea94b93a37
|
Fixes for unet
|
2020-11-15 10:38:33 -07:00 |
|
James Betker
|
89f56b2091
|
Fix another import
|
2020-11-14 22:10:45 -07:00 |
|
James Betker
|
9af049c671
|
Import fix for unet
|
2020-11-14 22:09:18 -07:00 |
|
James Betker
|
5cade6b874
|
Move stylegan2 around, bring in unet
|
2020-11-14 22:04:48 -07:00 |
|
James Betker
|
4c6b14a3f8
|
Allow extract_square_images to work on multiple images
|
2020-11-14 20:24:05 -07:00 |
|
James Betker
|
125cb16dce
|
Add a FID evaluator for stylegan with structural guidance
|
2020-11-14 20:16:07 -07:00 |
|
James Betker
|
c9258e2da3
|
Alter how structural guidance is given to stylegan
|
2020-11-14 20:15:48 -07:00 |
|
James Betker
|
3397c83447
|
Merge remote-tracking branch 'origin/gan_lab' into gan_lab
|
2020-11-14 09:30:09 -07:00 |
|
James Betker
|
423ee7cb90
|
Allow attention to be specified for stylegan2
|
2020-11-14 09:29:53 -07:00 |
|
James Betker
|
ec621c69b5
|
Fix train bug
|
2020-11-14 09:29:08 -07:00 |
|
James Betker
|
cdc5ac30e9
|
oddity
|
2020-11-13 20:11:57 -07:00 |
|
James Betker
|
f406a5dd4c
|
Mods to support stylegan2 in SR mode
|
2020-11-13 20:11:50 -07:00 |
|
James Betker
|
9c3d0b7560
|
Merge remote-tracking branch 'origin/gan_lab' into gan_lab
|
2020-11-13 20:10:47 -07:00 |
|
James Betker
|
67bf55495b
|
Allow hq_batched_key to be specified
|
2020-11-13 20:10:12 -07:00 |
|
James Betker
|
0b96811611
|
Fix another issue with gpu ids getting thrown all over hte place
|
2020-11-13 20:05:52 -07:00 |
|
James Betker
|
c47925ae34
|
New image extractor utility
|
2020-11-13 11:04:03 -07:00 |
|