training your own gan

frankligy · Mar 19, 2021 · 2b57cee · 2b57cee
1 parent e939d1d
commit 2b57cee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -9,7 +9,7 @@ The repository for building the DeepImmuno web server: https://github.com/frankl
 
 - Please refer to [DeepImmuno-GAN](#deepimmuno-gan) if you want to generate immunogenic peptide
 
-- Please refer to [Train your own GAN]() if you want to generate peptides with customized features/properties.
+- Please refer to [Train your own GAN](https://github.com/frankligy/DeepImmuno/tree/main/extension) if you want to generate peptides with customized features/properties.
 
 Enjoy and don't hesitate to ask me questions (contact at the bottom), I will be responsive! Feel free to raise an issue on github page!
 

diff --git a/extension/README.md b/extension/README.md
@@ -41,7 +41,7 @@ The examples can be found in `./gan_result` folder
 
 - In addtion to the basic dependencies shown in main [README.md](https://github.com/frankligy/DeepImmuno) file, you probably need `matplotlib` as well to draw the diagnostic plots.
 
-- We hard-coded the `batch_size=64`, so we hope your training data is at least more than 64 instances. Having said that, you can change this setting at your will.
+- We hard-coded the `batch_size=64`, so we hope your training data is at least more than 64 instances. Having said that, [you can change this setting at your will](https://github.com/frankligy/DeepImmuno/blob/main/extension/deepimmuno-gan-train.py/#L390).
 
 - we advise to set the `epoch` as 100 or more to assure a good training performance.