nodes/SayanoAI-Comfy-RVC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improved balancer + loss logic

Browse Source
This commit is contained in:
SayanoAI
2024-10-03 19:20:55 -04:00
parent 427fb7521d
commit 784767c8e0
4 changed files with 120 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_nodes/rvc_nodes.py
+2 -2
View File
@@ -419,7 +419,7 @@ class RVCTrainModelNode:
train_index=("BOOLEAN",{"default": True}),
retrain=("BOOLEAN",{"default": False}),
save_best_model=("BOOLEAN",{"default": True}),
best_model_threshold=("INT",dict(default=30,min=10,max=50)),
best_model_threshold=("INT",dict(default=10,min=1,max=100)),
log_every_epoch=("FLOAT",dict(default=1.,min=0.,max=2.,step=.1)),
num_workers=("INT",dict(default=1,min=1,max=16))
)
@@ -448,7 +448,7 @@ class RVCTrainModelNode:
train_index=True,
retrain=False,
save_best_model=True,
best_model_threshold=50,
best_model_threshold=10,
log_every_epoch=1.,
num_workers=1):