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
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.