James Betker
|
2c65b6b28e
|
More mods to support styledsr
|
2021-01-04 11:32:28 -07:00 |
|
James Betker
|
2225fe6ac2
|
Undo lucidrains changes for new discriminator
This "new" code will live in the styledsr directory from now on.
|
2021-01-04 10:57:09 -07:00 |
|
James Betker
|
40ec71da81
|
Move styled_sr into its own folder
|
2021-01-04 10:54:34 -07:00 |
|
James Betker
|
5916f5f7d4
|
Misc fixes
|
2021-01-04 10:53:53 -07:00 |
|
James Betker
|
4d8064c32c
|
Modifications to allow partially trained stylegan discriminators to be used
|
2021-01-03 16:37:18 -07:00 |
|
James Betker
|
5e7ade0114
|
ImageFolderDataset - corrupt lq images alongside each other
|
2021-01-03 16:36:38 -07:00 |
|
James Betker
|
ce6524184c
|
Do the last commit but in a better way
|
2021-01-02 22:24:12 -07:00 |
|
James Betker
|
edf9c38198
|
Make ExtensibleTrainer set the starting step for the LR scheduler
|
2021-01-02 22:22:34 -07:00 |
|
James Betker
|
bdbab65082
|
Allow optimizers to train separate param groups, add higher dimensional VGG discriminator
Did this to support training 512x512px networks off of a pretrained 256x256 network.
|
2021-01-02 15:10:06 -07:00 |
|
James Betker
|
193cdc6636
|
Move discriminators to the create_model paradigm
Also cleans up a lot of old discriminator models that I have no intention
of using again.
|
2021-01-01 15:56:09 -07:00 |
|
James Betker
|
7976a5825d
|
srfid is incorrectly labeled
|
2021-01-01 13:00:59 -07:00 |
|
James Betker
|
f39179e85a
|
styled_sr: fix bug when using initial_stride
|
2021-01-01 12:13:21 -07:00 |
|
James Betker
|
913fc3b75e
|
Need init to pick up styled_sr
|
2021-01-01 12:10:32 -07:00 |
|
James Betker
|
aae65e6ed8
|
Mods to byol_resnet_playground for large batches
|
2021-01-01 11:59:54 -07:00 |
|
James Betker
|
e992e18767
|
Add initial_stride term to style_sr
Also fix fid and a networks.py issue.
|
2021-01-01 11:59:36 -07:00 |
|
James Betker
|
9864fe4c04
|
Fix for train.py
|
2021-01-01 11:59:00 -07:00 |
|
James Betker
|
e214e6ce33
|
Styled SR model
|
2020-12-31 20:54:18 -07:00 |
|
James Betker
|
0eb1f4dd67
|
Revert "Get rid of CUDA_VISIBLE_DEVICES"
It is actually necessary for training in distributed mode. Only
do it then.
|
2020-12-31 10:31:40 -07:00 |
|
James Betker
|
8de5a02a48
|
byol_resnet_playground
Similar to the spinenet playground, but tinkers with resnet instead
|
2020-12-31 10:15:04 -07:00 |
|
James Betker
|
8f18b2709e
|
Get rid of CUDA_VISIBLE_DEVICES
It is not clear to me what the purpose of this is, but it has recently
started causing failures.
|
2020-12-31 10:13:58 -07:00 |
|
James Betker
|
1de1fa30ac
|
Disable refs and centers altogether in single_image_dataset
I suspect that this might be a cause of failures on parallel datasets.
Plus it is unnecessary computation.
|
2020-12-31 10:13:24 -07:00 |
|
James Betker
|
8f0984cacf
|
Add sr_fid evaluator
|
2020-12-30 20:18:58 -07:00 |
|