kregmi/vte

video trajectory estimation

vte's Issues

BDD-GSV dataset

Hi Krishna,

Thanks for your great work and for sharing your codes. I'm wondering if you could share the whole BDD-GSV dataset so that I can directly run your codes and make some experiments.

Best regards,
Dong Yuan

model.pool not defined in test_bdd.py

Hello Dr Krishna,
It seems model.pool was not defined anywhere before using in the encoder. How can I change it. As you can see model.pool was being used without defining.

        encoder = nn.Sequential(*layers)
	model = nn.Module()
	model.add_module('encoder', encoder)


	isParallel = False
	if opt.nGPU > 1 and torch.cuda.device_count() > 1:
		model.encoder = nn.DataParallel(model.encoder)
		if opt.mode.lower() != 'cluster':
			model.pool = nn.DataParallel(model.pool)
		isParallel = True