Sovit Ranjan Rath
Sovit Ranjan Rath
You use the `--imgsz` command line argument to control the image size.
There are a lot actually. I will recommend going through all the command line arguments in the `train.py` file. Especially, the ones that handle arguments. `--mosaic` `--use-train-aug`
I think creating a separate `metrics.py` inside the `utils` directory will be the best approach.
Oh, if you wish to add that to the evaluation part after training only, then you can directly modify the `eval.py` script in the root directory.
It is basically saying that the list and tensor data types do not have a `replace` attribute. I think string data types have `replace` but that may not be an...
I need to take a look how to structure the additional changes.
Yes, it is possible. I intend to update the README with a lot of things including this after I merge your PR. For the time being, please refer to this....
I will check the logging issue that you mention above. Regarding the best model issues, can you elaborate on whether you are trying to resume training or using the best...
Thanks for informing this. I will surely check this out. I am not sure how the head will differ on 4 different GPUs compared to a single GPU. But I...
The Vision Transformer models are a work in progress and they depend on this library of mine https://github.com/sovit-123/vision_transformers It is recommended that you don't use the models now. I will...