Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training own dataset #26

Closed
junyang-cynopsis opened this issue May 27, 2019 · 4 comments
Closed

Training own dataset #26

junyang-cynopsis opened this issue May 27, 2019 · 4 comments

Comments

@junyang-cynopsis
Copy link

junyang-cynopsis commented May 27, 2019

Hi I want to train my own dataset.

I received the following error
The directory name is invalid: 'weights\rrdn-C4-D3-G64-G064-T10-x2\2019-05-27_23:41'

Any reasons why?

@junyang-cynopsis junyang-cynopsis changed the title Training own dataset Training own dataset, directory name is invalid May 27, 2019
@junyang-cynopsis
Copy link
Author

junyang-cynopsis commented May 27, 2019

Okay I got it, apparently windows cant create folders with semicolon in its filename, which appears in the timestamp. I removed the timestamp(or semicolon) and it works now

@junyang-cynopsis junyang-cynopsis changed the title Training own dataset, directory name is invalid Training own dataset May 27, 2019
@junyang-cynopsis
Copy link
Author

I have an issue now, the code doesnt save my hdf5 weights after end of epochs training, any reasons why?

@junyang-cynopsis
Copy link
Author

Okay I got it too

So we need to set the monitor metrics to be specifically one of the following keys, cant use the default "val_generator_PSNR". ie example use "val_generator_PSNR_Y"

{'train_d_real_loss': 0.060769938, 'train_d_real_acc': 1.0, 'train_d_fake_loss': 0.027885916, 'train_d_fake_acc': 1.0, 'train_loss': 0.18021318, 'train_generator_loss': 0.1242274, 'train_discriminator_loss': 4.2881703, 'train_feature_extractor_loss': 2.6231012, 'train_generator_PSNR_Y': 17.87545}

{'val_loss': 0.14819694967009128, 'val_generator_loss': 0.1413433525711298, 'val_discriminator_loss': 0.6808462247252465, 'val_feature_extractor_loss': 2.6356906309723853, 'val_generator_PSNR_Y': 17.35666718482971}

@cfrancesco
Copy link
Collaborator

Okay I got it, apparently windows cant create folders with semicolon in its filename, which appears in the timestamp. I removed the timestamp(or semicolon) and it works now

Cool, I was not aware of this issue, thank you for pointing it out. I will look into a fix.

Okay I got it too

So we need to set the monitor metrics to be specifically one of the following keys, cant use the default "val_generator_PSNR". ie example use "val_generator_PSNR_Y"}

There needs to be at least one monitored metric amongst the available metrics. If not, it should only save the weights during the fallback saves. I will double check the behavior.
I am strongly considering to remove a weight saving logic in favor of a "save every epoch, mark if it is best on some metric" kind of logic, especially due to the difficulties while evaluating the perceptual quality (when GANS and VGG loss are introduced).
Thank you for your effort!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants