James Betker
|
e838c6e75b
|
Rosinality stylegan2 port
|
2020-12-17 14:18:46 -07:00 |
|
James Betker
|
fc376d34b2
|
Spinenet with logits head
|
2020-12-15 17:16:19 -07:00 |
|
James Betker
|
26ceca68c0
|
BYOL with structure!
|
2020-12-10 15:07:35 -07:00 |
|
James Betker
|
97ff25a086
|
BYOL!
Man, is there anything ExtensibleTrainer can't train? :)
|
2020-12-08 13:07:53 -07:00 |
|
James Betker
|
711780126e
|
Cleanup
|
2020-12-03 23:42:51 -07:00 |
|
James Betker
|
06d1c62c5a
|
iGPT support!
Sweeeeet
|
2020-12-03 15:32:21 -07:00 |
|
James Betker
|
2e0bbda640
|
Remove unused archs
|
2020-12-01 11:10:48 -07:00 |
|
James Betker
|
a1c8300052
|
Add mdcn
|
2020-11-30 16:14:21 -07:00 |
|
James Betker
|
1e0f69e34b
|
extra_conv in gn discriminator, multiframe support in rrdb.
|
2020-11-29 15:39:50 -07:00 |
|
James Betker
|
da604752e6
|
Misc RRDB changes
|
2020-11-29 12:21:31 -07:00 |
|
James Betker
|
a1d4c9f83c
|
multires rrdb work
|
2020-11-28 14:35:46 -07:00 |
|
James Betker
|
929cd45c05
|
Fix for RRDB scale
|
2020-11-27 21:37:10 -07:00 |
|
James Betker
|
4ab49b0d69
|
RRDB disc work
|
2020-11-27 12:03:08 -07:00 |
|
James Betker
|
fd356580c0
|
Play with lambdas
|
2020-11-26 20:30:55 -07:00 |
|
James Betker
|
cb045121b3
|
Expose srflow rrdb
|
2020-11-24 13:20:20 -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
|
5ccdbcefe3
|
srflow_orig integration
|
2020-11-19 23:47:24 -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
|
e587d549f7
|
Fix circular imports
|
2020-11-15 11:32:35 -07:00 |
|
James Betker
|
5cade6b874
|
Move stylegan2 around, bring in unet
|
2020-11-14 22:04:48 -07:00 |
|
James Betker
|
423ee7cb90
|
Allow attention to be specified for stylegan2
|
2020-11-14 09:29:53 -07:00 |
|
James Betker
|
f406a5dd4c
|
Mods to support stylegan2 in SR mode
|
2020-11-13 20:11:50 -07:00 |
|
James Betker
|
2d3449d7a5
|
stylegan2 in ml art school!
|
2020-11-12 15:42:05 -07:00 |
|
James Betker
|
b4136d766a
|
Back to pyramids, no rrdb
|
2020-11-11 13:40:24 -07:00 |
|
James Betker
|
42a97de756
|
Convert PyramidRRDBDisc to RRDBDisc
Had numeric stability issues. This probably makes more sense anyways.
|
2020-11-11 12:14:14 -07:00 |
|
James Betker
|
72762f200c
|
PyramidRRDB net
|
2020-11-11 11:25:49 -07:00 |
|
James Betker
|
a1760f8969
|
Adapt srg2 for video
|
2020-11-10 16:16:41 -07:00 |
|
James Betker
|
6a2fd5f7d0
|
Lots of new discriminator nets
|
2020-11-10 16:06:54 -07:00 |
|
James Betker
|
9e2c96ad5d
|
More latent work
|
2020-11-07 20:38:56 -07:00 |
|
James Betker
|
0cf52ef52c
|
latent work
|
2020-11-06 20:38:23 -07:00 |
|
James Betker
|
fd6cdba88f
|
RRDB with latent
|
2020-11-05 10:04:17 -07:00 |
|
James Betker
|
df47d6cbbb
|
More work in support of training flow networks in tandem with generators
|
2020-11-04 18:07:48 -07:00 |
|
James Betker
|
dcfe994fee
|
Add standalone srg2_classic
Trying to investigate how I was so misguided. I *thought* srg2 was considerably
better than RRDB in performance but am not actually seeing that.
|
2020-10-31 20:55:34 -06:00 |
|
James Betker
|
7303d8c932
|
Add psnr approximator
|
2020-10-31 11:08:55 -06:00 |
|
James Betker
|
7d38381d46
|
Add scaling to rrdb
|
2020-10-29 09:48:10 -06:00 |
|
James Betker
|
607ff3c67c
|
RRDB with bypass
|
2020-10-29 09:39:45 -06:00 |
|
James Betker
|
9848f4c6cb
|
Add teco_resgen
|
2020-10-27 20:59:55 -06:00 |
|
James Betker
|
8202ee72b9
|
Re-add original SPSR_arch
|
2020-10-27 11:00:38 -06:00 |
|
James Betker
|
231137ab0a
|
Revert RRDB back to original model
|
2020-10-27 10:25:31 -06:00 |
|
James Betker
|
9c3d059ef0
|
Updates to be able to train flownet2 in ExtensibleTrainer
Only supports basic losses for now, though.
|
2020-10-24 11:56:39 -06:00 |
|
James Betker
|
1dbcbfbac8
|
Restore ChainedEmbeddingGenWithStructure
Still using this guy, after all
|
2020-10-24 11:54:52 -06:00 |
|
James Betker
|
7a75d10784
|
Arch cleanup
|
2020-10-23 09:35:33 -06:00 |
|
James Betker
|
646d6a621a
|
Support 4x zoom on ChainedEmbeddingGen
|
2020-10-23 09:25:58 -06:00 |
|
James Betker
|
1ef559d7ca
|
Add a ChainedEmbeddingGen which can be simueltaneously used with multiple training paradigms
|
2020-10-21 22:21:51 -06:00 |
|
James Betker
|
dca5cddb3b
|
Add bypass to ChainedEmbeddingGen
|
2020-10-21 11:07:45 -06:00 |
|
James Betker
|
7df378a944
|
Remove separated vgg discriminator
Checkpointing happens inline instead. Was a dumb idea..
Also fixes some loss reporting issues.
|
2020-10-18 12:10:24 -06:00 |
|
James Betker
|
6a0d5f4813
|
Add a checkpointable discriminator
|
2020-10-18 09:57:47 -06:00 |
|