NikkeDoy/SISU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚗️ | Experimenting with larger model architecture.

Browse Source
This commit is contained in:
NikkeDoy
2025-01-08 15:33:18 +02:00
parent 89f8c68986
commit f615b39ded
3 changed files with 53 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
training.py
View File

@ -175,16 +175,15 @@ def start_training():
if generator_epoch % 10 == 0:
print(f"Saved epoch {generator_epoch}!")
torchaudio.save(f"./output/epoch-{generator_epoch}-audio-crap.wav", low_quality_audio[0][0].cpu(), high_quality_audio[1]) # <-- Because audio clip was resampled in data.py from what ever that low_quality had to high_quality
torchaudio.save(f"./output/epoch-{generator_epoch}-audio-crap.wav", low_quality_audio[0][0].cpu(), high_quality_audio[1]) # <-- Because audio clip was resampled in data.py from original to crap and to original again.
torchaudio.save(f"./output/epoch-{generator_epoch}-audio-ai.wav", ai_enhanced_audio[0][0].cpu(), ai_enhanced_audio[1])
torchaudio.save(f"./output/epoch-{generator_epoch}-audio-orig.wav", high_quality_audio[0][0].cpu(), high_quality_audio[1])
if generator_epoch % 50 == 0:
torch.save(discriminator.state_dict(), f"models/epoch-{generator_epoch}-discriminator.pt")
torch.save(generator.state_dict(), f"models/epoch-{generator_epoch}-generator.pt")
torch.save(discriminator.state_dict(), f"models/current-epoch-discriminator.pt")
torch.save(generator.state_dict(), f"models/current-epoch-generator.pt")
torch.save(discriminator.state_dict(), "models/epoch-500-discriminator.pt")
torch.save(generator.state_dict(), "models/epoch-500-generator.pt")
torch.save(discriminator.state_dict(), "models/epoch-5000-discriminator.pt")
torch.save(generator.state_dict(), "models/epoch-5000-generator.pt")
print("Training complete!")
start_training()