Open TheKursta opened 2 years ago
Hi!
When I try to execute model.summary() after loading weights in eval.py line 77, I get following error:
"" summary() missing 1 required positional argument: 'input_shape' ""
Is there a particular pipeline needed to be executed to have model architecture be easily printable? I would like to remove final classification head and append my own for fine-tuning on different dataset.
Hi!
When I try to execute model.summary() after loading weights in eval.py line 77, I get following error:
"" summary() missing 1 required positional argument: 'input_shape' ""
Is there a particular pipeline needed to be executed to have model architecture be easily printable? I would like to remove final classification head and append my own for fine-tuning on different dataset.