fc_model.py.update_input_length: .to(device) makes tests fail but is necessary to enable input size change
fc_model.py.update_input_length:
.to(device) makes tests fail but is necessary to enable input size change