Clone/DIS
1
0
Fork 0
mirror of https://github.com/xuebinqin/DIS.git synced 2024-11-26 08:43:17 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

dis v2

Browse Source
This commit is contained in:
Xuebin Qin 2022-07-17 16:19:20 -07:00
parent b9e31bdc26
commit 85a9a1be60
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
.DS_Store vendored
View File

Binary file not shown.

4
IS-Net/train_valid_inference_main.py
View File

@ -709,9 +709,9 @@ if __name__ == "__main__":
print("building model...")
hypar["model"] = ISNetDIS() #U2NETFASTFEATURESUP()
hypar["early_stop"] = 20 ## stop the training when no improvement in the past 20 validation periods, smaller numbers can be used here e.g., 5 or 10.
hypar["model_save_fre"] = 20 ## valid and save model weights every 2000 iterations
hypar["model_save_fre"] = 2000 ## valid and save model weights every 2000 iterations
hypar["batch_size_train"] = 2 ## batch size for training
hypar["batch_size_train"] = 8 ## batch size for training
hypar["batch_size_valid"] = 1 ## batch size for validation and inferencing
print("batch size: ", hypar["batch_size_train"])