Harry/tfworld-hackathon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated .gitignore

Browse Source
This commit is contained in:
Harry Stuart 2020-01-03 12:24:29 +11:00
parent 26aa9e4b84
commit a48249554c
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.gitignore vendored
View File

@ -243,3 +243,7 @@ paket-files/
# JetBrains Rider # JetBrains Rider
.idea/ .idea/
*.sln.iml *.sln.iml
# Custom
media/
logs/

10
scripts/WGAN-GP.py
View File

@ -29,6 +29,16 @@ Fs = 16000
DATA_DIR = r"D:\ML_Datasets\violin" DATA_DIR = r"D:\ML_Datasets\violin"
INSTRUMENT = "violin" INSTRUMENT = "violin"
print("Creating necessary directories")
paths = ["logs/train",
f"models/{INSTRUMENT}/js",
f"output/{INSTRUMENT}",]
for path in paths:
if not os.path.exists(os.path.join(os.getcwd(), path)):
os.makedirs(path)
# Define class that contains GAN infrastructure # Define class that contains GAN infrastructure
class GAN: class GAN:
def __init__(self, model_dims=MODEL_DIMS, num_samples=NUM_SAMPLES, def __init__(self, model_dims=MODEL_DIMS, num_samples=NUM_SAMPLES,