Skip to content

Commit

Permalink
Update train_C2FNet.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ChengxiHAN authored Mar 26, 2024
1 parent 4da3078 commit e8eeae3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions train_C2FNet.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ def train1(train_loader, val_loader, Eva_train,Eva_train2, Eva_val,Eva_val2,
parser.add_argument('--lr', type=float, default=5e-4, help='learning rate')
parser.add_argument('--batchsize', type=int, default=16, help='training batch size') #修改这里!!!
parser.add_argument('--trainsize', type=int, default=256, help='training dataset size')
parser.add_argument('--train_ratio', type=float, default=0.05, help='Proportion of the labeled images')#修改这里!!!
parser.add_argument('--train_ratio', type=float, default=1, help='Proportion of the labeled images')#修改这里!!!
parser.add_argument('--clip', type=float, default=0.5, help='gradient clipping margin')
parser.add_argument('--decay_rate', type=float, default=0.1, help='decay rate of learning rate')
parser.add_argument('--decay_epoch', type=int, default=50, help='every n epochs decay learning rate')
Expand All @@ -219,7 +219,7 @@ def train1(train_loader, val_loader, Eva_train,Eva_train2, Eva_val,Eva_val2,
parser.add_argument('--save_path', type=str, default='./output/C2FNet/WHU/') # 全监督的模型保存路径!!

opt = parser.parse_args()
print('labeled ration=0.05,Ablation现在半监督损失函数系数为:0.2!')
print('labeled ration=1,Ablation现在半监督损失函数系数为:0.2!')

# set the device for training
if opt.gpu_id == '0':
Expand Down Expand Up @@ -308,7 +308,7 @@ def train1(train_loader, val_loader, Eva_train,Eva_train2, Eva_val,Eva_val2,
# 可以先全用有标签的训练几个epoch,再进行半监督训练 !!!!
# if epoch<5: #默认的为5,测试10,15,20
# use_ema=False
# # print('labeled ration=0.05,Ablation现在监督训练的次数为:20!')
# # print('labeled ration=1,Ablation现在监督训练的次数为:20!')
# else:
# use_ema=True

Expand Down

0 comments on commit e8eeae3

Please sign in to comment.