the 900m values in self.linear1 = nn.Linear(900, 128) is different from the original script and is based on the tensor's size.
Doing so, the training doesn't throw an error but it doesn't learn.
is there anything specific I need to do when I want to change the resolution?
Hello,
I have been trying to train ERF_Net on CULane and I have encountered some issues:
first of all, I change the parameters image_height and width to half the original value. here is my train script:
!python3 -u train_erfnet.py CULane ERFNet ./train_gt ./val_gt \ --lr 0.01 \ --gpus 0 \ --resume ./pretrained/ERFNet_pretrained.tar \ -j 4 \ -b 32 \ --epochs 20 \ --img_height 104 \ --img_width 488 \ 2>&1|tee train_erfnet_culane.log
my first issue is that I need to change the size of two arrays in the original script erfnet.py otherwise it throws an error:
`class Lane_exist(nn.Module): def init(self, num_output): super().init()
the 900m values in self.linear1 = nn.Linear(900, 128) is different from the original script and is based on the tensor's size. Doing so, the training doesn't throw an error but it doesn't learn.
is there anything specific I need to do when I want to change the resolution?