Skip to content

Commit 92f4457

Browse files
committed
add gitignore and update train.py
1 parent 17d9d46 commit 92f4457

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

.gitignore

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,10 @@
1+
# ignore map, miou, datasets
2+
map_out/
3+
miou_out/
4+
VOCdevkit/
5+
datasets/
6+
Medical_Datasets/
7+
18
# Byte-compiled / optimized / DLL files
29
__pycache__/
310
*.py[cod]

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@
239239
#-------------------------------------------------------------------#
240240
# 判断当前batch_size与64的差别,自适应调整学习率
241241
#-------------------------------------------------------------------#
242-
nbs = 64 if not Freeze_Train else 32 # nominal batch size
242+
nbs = 64
243243
Init_lr = max(batch_size / nbs * Init_lr, 1e-4)
244244
Min_lr = max(batch_size / nbs * Min_lr, 1e-6)
245245

0 commit comments

Comments
 (0)