Switching to saving samples and checkpoints based on kimgs #21

duskvirkus · 2021-08-09T11:52:30Z

fixed tensor device bug in EqualLinear.forward() method
added 3 training integration tests (from scratch, ffhq, and custom resume)
switched to using wget with digital ocean hosted pretrained models
moved ci files out of repository
added create_sample_grid_vectors.py script
added KimgSaverCallback which saves samples and checkpoints based on kimgs
removed --n_samples argument
added --save_sample_every_kimgs, --save_checkpoint_every_kimgs, --start_kimg_count, --stop_training_at_kimgs and --sample_grid arguments

duskvirkus added 3 commits August 9, 2021 06:37

recommiting code changes

c64aba4

adding ci files to install

a193b00

forgot sample grid

9dc4696

duskvirkus added this to the v1.0.0 milestone Aug 9, 2021

duskvirkus added 10 commits August 9, 2021 11:59

changing to apt-get

343a100

removing wget install

350af28

fixing path on ci

938215d

making sure to use python3

cc830dc

how about that

d0b4263

trying this instead

d023bff

missed a dataset path

1cd7d40

updating paths

fdf62cf

forgot to save file

8526091

maybe this will work?

58539f0

duskvirkus merged commit acf8245 into devel Aug 16, 2021

duskvirkus deleted the kimgs branch August 16, 2021 04:14

Provide feedback