What is the difference between Stage 2 and Stage 3?

[longmalongma]

What is the difference between Stage 2 and Stage 3? Do I need to execute --stage 3 after --stage 2?

[hkchengrex]

0 -> 1 -> 2 = With BL30K
0 -> 3 = Without BL30K

STCN/util/hyper_para.py

Lines 67 to 80 in 5f11ad4

	elif self.args['stage'] == 2:
	# 300K main training for after BL30K
	self.args['lr'] = none_or_default(self.args['lr'], 1e-5)
	self.args['batch_size'] = none_or_default(self.args['batch_size'], 4)
	self.args['iterations'] = none_or_default(self.args['iterations'], 300000)
	self.args['steps'] = none_or_default(self.args['steps'], [250000])
	self.args['single_object'] = False
	elif self.args['stage'] == 3:
	# 150K main training for after static image pretraining
	self.args['lr'] = none_or_default(self.args['lr'], 1e-5)
	self.args['batch_size'] = none_or_default(self.args['batch_size'], 4)
	self.args['iterations'] = none_or_default(self.args['iterations'], 150000)
	self.args['steps'] = none_or_default(self.args['steps'], [125000])
	self.args['single_object'] = False

[longmalongma]

0 -> 1 -> 2 = With BL30K
0 -> 3 = Without BL30K

STCN/util/hyper_para.py

Lines 67 to 80 in 5f11ad4

	elif self.args['stage'] == 2:
	# 300K main training for after BL30K
	self.args['lr'] = none_or_default(self.args['lr'], 1e-5)
	self.args['batch_size'] = none_or_default(self.args['batch_size'], 4)
	self.args['iterations'] = none_or_default(self.args['iterations'], 300000)
	self.args['steps'] = none_or_default(self.args['steps'], [250000])
	self.args['single_object'] = False
	elif self.args['stage'] == 3:
	# 150K main training for after static image pretraining
	self.args['lr'] = none_or_default(self.args['lr'], 1e-5)
	self.args['batch_size'] = none_or_default(self.args['batch_size'], 4)
	self.args['iterations'] = none_or_default(self.args['iterations'], 150000)
	self.args['steps'] = none_or_default(self.args['steps'], [125000])
	self.args['single_object'] = False

Ok, thanks, I got it1