Commit Graph

  • b3a2c21250 err2 James Betker 2020-07-10 22:52:02 -0600
  • 716433db1f err1 James Betker 2020-07-10 22:50:56 -0600
  • ef9f1307eb Sometimes don't use compression artifacts James Betker 2020-07-10 22:25:53 -0600
  • 0b7193392f Implement unet disc James Betker 2020-07-10 16:16:03 -0600
  • 812c684f7d Update pixgan swap algorithm James Betker 2020-07-10 15:56:14 -0600
  • 33ca3832e1 Move ExpansionBlock to arch_util James Betker 2020-07-10 15:53:41 -0600
  • 5e8b52f34c Misc changes James Betker 2020-07-10 09:45:34 -0600
  • 5f2c722a10 SRG2 revival James Betker 2020-07-09 17:34:51 -0600
  • 12da993da8 More fixes... James Betker 2020-07-08 22:07:09 -0600
  • 7d6eb28b87 More fixes James Betker 2020-07-08 22:00:57 -0600
  • b2507be13c Fix up pixgan loss and pixdisc James Betker 2020-07-08 21:27:48 -0600
  • 26a4a66d1c Bug fixes and new gan mechanism James Betker 2020-07-08 17:40:26 -0600
  • 4305be97b4 Update log metrics James Betker 2020-07-07 15:33:22 -0600
  • 8a4eb8241d SRG3 work James Betker 2020-07-07 13:46:40 -0600
  • 0acad81035 More SRG2 adjustments.. James Betker 2020-07-06 22:40:40 -0600
  • 086b2f0570 More bugs James Betker 2020-07-06 22:28:07 -0600
  • d4d4f85fc0 Bug fixes James Betker 2020-07-06 22:25:40 -0600
  • 3c31bea1ac SRG2 architectural changes James Betker 2020-07-06 22:22:29 -0600
  • 9a1c3241f5 Switch discriminator to groupnorm James Betker 2020-07-06 20:59:59 -0600
  • 60c6352843 Misc James Betker 2020-07-06 20:44:07 -0600
  • 6beefa6d0c PixDisc - Add two more levels of losses coming from this gen at higher resolutions James Betker 2020-07-06 11:15:52 -0600
  • 2636d3b620 Fix assertion error James Betker 2020-07-06 09:23:53 -0600
  • 8f92c0a088 Interpolate attention well before softmax James Betker 2020-07-06 09:18:30 -0600
  • 72f90cabf8 More pixdisc fixes James Betker 2020-07-05 22:03:16 -0600
  • 909007ee6a Add G_warmup James Betker 2020-07-05 21:58:35 -0600
  • a47a5dca43 Fix pixdisc bug James Betker 2020-07-05 21:49:09 -0600
  • d0957bd7d4 Alter weight initialization for transformation blocks James Betker 2020-07-05 17:32:46 -0600
  • 16d1bf6dd7 Replace ConvBnRelus in SRG2 with Silus James Betker 2020-07-05 17:28:00 -0600
  • 10f7e49214 Add ConvBnSilu to replace ConvBnRelu James Betker 2020-07-05 13:39:08 -0600
  • 9934e5d082 Move SRG1 to identical to new James Betker 2020-07-05 08:49:34 -0600
  • 416538f31c SRG1 conjoined except ConvBnRelu James Betker 2020-07-05 08:44:17 -0600
  • c58c2b09ca Back to remove all biases (looks like a ConvBnRelu made its way in..) James Betker 2020-07-04 22:41:02 -0600
  • 86cda86e94 Re-add biases, also add new init James Betker 2020-07-04 22:24:42 -0600
  • b03741f30e Remove all biases from generator James Betker 2020-07-04 22:19:55 -0600
  • 726e946e79 Turn BN off in SRG1 James Betker 2020-07-04 14:51:27 -0600
  • 0ee39d419b OrderedDict not needed James Betker 2020-07-04 14:09:27 -0600
  • 9048105b72 Break out SRG1 as separate network James Betker 2020-07-04 13:28:50 -0600
  • 188de5e15a Misc changes James Betker 2020-07-04 13:22:50 -0600
  • 510b2f887d Remove RDB from srg2 James Betker 2020-07-03 22:31:20 -0600
  • 77d3765364 Fix new feature loss calc James Betker 2020-07-03 22:20:13 -0600
  • ed6a15e768 Add feature to dataset which allows it to force images to be a certain size. James Betker 2020-07-03 15:19:16 -0600
  • da4335c25e Add a feature-based validation test James Betker 2020-07-03 15:18:57 -0600
  • 703dec4472 Add SpineNet & integrate with SRG James Betker 2020-07-03 12:07:31 -0600
  • 3ed7a2b9ab Move ConvBnRelu/Lelu to arch_util James Betker 2020-07-03 12:06:38 -0600
  • ea9c6765ca Move train imports into init_dist James Betker 2020-07-02 15:11:21 -0600
  • e9ee67ff10 Integrate RDB into SRG James Betker 2020-07-01 17:19:55 -0600
  • 6ac6c95177 Fix scaling bug James Betker 2020-07-01 16:42:27 -0600
  • 30653181ba Experiment: get rid of post_switch_conv James Betker 2020-07-01 16:30:40 -0600
  • 17191de836 Experiment: bring initialize_weights back again James Betker 2020-07-01 15:57:55 -0600
  • d1d573de07 Experiment: new init and post-switch-conv James Betker 2020-07-01 15:25:54 -0600
  • 480d1299d7 Remove RRDB with switching James Betker 2020-07-01 12:08:32 -0600
  • e2398ac83c Experiment: revert initialization changes James Betker 2020-07-01 12:08:09 -0600
  • 78276afcaa Experiment: Back to lelu James Betker 2020-07-01 11:43:25 -0600
  • b945021c90 SRG v2 - Move to Relu, rely on Module-based initialization James Betker 2020-07-01 11:33:32 -0600
  • ee6443ad7d Add numeric stability computation script James Betker 2020-07-01 11:30:34 -0600
  • c0bb123504 Misc changes James Betker 2020-07-01 11:28:23 -0600
  • 604763be68 NSG r7 James Betker 2020-07-01 09:54:29 -0600
  • 87f1e9c56f Invert ResGen2 to operate in LR space James Betker 2020-06-30 20:57:40 -0600
  • e07d8abafb NSG rev 6 James Betker 2020-06-30 20:47:26 -0600
  • 3ce1a1878d NSG improvements (r5) James Betker 2020-06-30 16:59:57 -0600
  • 75f148022d Even more NSG improvements (r4) James Betker 2020-06-30 13:52:47 -0600
  • 773753073f More NSG improvements (v3) James Betker 2020-06-29 20:26:51 -0600
  • 4b82d0815d NSG improvements James Betker 2020-06-29 10:09:51 -0600
  • 978036e7b3 Add NestedSwitchGenerator James Betker 2020-06-28 21:21:57 -0600
  • 6f2bc36c61 Distill_torchscript mods James Betker 2020-06-27 08:28:09 -0600
  • db08dedfe2 Add recover_tensorboard_log James Betker 2020-06-27 08:26:57 -0600
  • c8a670842e Missed networks.py in last commit James Betker 2020-06-25 18:36:06 -0600
  • 407224eba1 Re-work SwitchedResgen2 James Betker 2020-06-25 18:17:05 -0600
  • 42a10b34ce Re-enable batch norm on switch processing blocks James Betker 2020-06-24 21:15:17 -0600
  • 4001db1ede Add ConfigurableSwitchComputer James Betker 2020-06-24 19:49:37 -0600
  • 83c3b8b982 Add parameterized noise injection into resgen James Betker 2020-06-23 10:16:02 -0600
  • 0584c3b587 Add negative_transforms switch to resgen James Betker 2020-06-23 09:41:12 -0600
  • dfcbe5f2db Add capability to place additional conv into discriminator James Betker 2020-06-23 09:40:33 -0600
  • bad33de906 Add simple resize to extract images James Betker 2020-06-23 09:39:51 -0600
  • 030648f2bc Remove batchnorms from resgen James Betker 2020-06-22 17:23:36 -0600
  • 68bcab03ae Add growth channel to switch_growths for flat networks James Betker 2020-06-22 10:40:16 -0600
  • 3b81712c49 Remove BN from transforms James Betker 2020-06-19 16:52:56 -0600
  • 61364ec7d0 Fix inverse temperature curve logic and add upsample factor James Betker 2020-06-19 09:18:30 -0600
  • 0551139b8d Fix resgen temperature curve below 1 James Betker 2020-06-18 16:08:07 -0600
  • efc80f041c Save & load amp state James Betker 2020-06-18 11:38:48 -0600
  • 2e3b6bad77 Log tensorboard directly into experiments directory James Betker 2020-06-18 11:33:02 -0600
  • 778e7b6931 Add a double-step to attention temperature James Betker 2020-06-18 11:29:31 -0600
  • d2d5e097d5 Add profiling to SRGAN for testing timings James Betker 2020-06-18 11:29:10 -0600
  • 45a900fafe Misc James Betker 2020-06-18 11:28:55 -0600
  • 59b0533b06 Fix attimage step size James Betker 2020-06-17 18:45:24 -0600
  • 645d0ca767 ResidualGen mods James Betker 2020-06-17 17:18:28 -0600
  • 6f8406fbdc Fixed ConfigurableSwitchedGenerator bug James Betker 2020-06-16 16:53:57 -0600
  • 7d541642aa Get rid of SwitchedResidualGenerator James Betker 2020-06-16 16:22:56 -0600
  • 379b96eb55 Output histograms with SwitchedResidualGenerator James Betker 2020-06-16 15:54:37 -0600
  • f8b67f134b Get proper contiguous view for backwards compatibility James Betker 2020-06-16 14:27:16 -0600
  • 2def96203e Mods to SwitchedResidualGenerator_arch James Betker 2020-06-16 14:19:12 -0600
  • 70c764b9d4 Create a configurable SwichedResidualGenerator James Betker 2020-06-16 13:24:07 -0600
  • df1046c318 New arch: SwitchedResidualGenerator_arch James Betker 2020-06-16 11:23:50 -0600
  • ddfd7f67a0 Get rid of biggan James Betker 2020-06-16 11:21:44 -0600
  • 0a714e8451 Fix initialization in mhead switched rrdb James Betker 2020-06-15 21:32:03 -0600
  • be7982b9ae Add skip heads to switcher James Betker 2020-06-14 12:46:54 -0600
  • 6c27ddc9b5 Misc James Betker 2020-06-14 11:03:02 -0600
  • 6c0e9f45c7 Add GPU mem tracing module James Betker 2020-06-14 11:02:54 -0600
  • 48532a0a8a Fix initial_stride on lowdim models James Betker 2020-06-14 11:02:16 -0600
  • 532704af40 Multiple modifications for experimental RRDB architectures James Betker 2020-06-13 11:37:27 -0600