Open JackGit opened 2 months ago
./final_model_csv
is where we store our final models. Please remove the command for creating this folder when building docker.
Before running the training, please make sure to download the ImageNet-1K pre-trained weights for the two models from the following links and place them in the./pre_model
directory:
RepLKNet: https://drive.google.com/file/d/1vo-P3XB6mRLUeDzmgv90dOu73uCeLfZN/view?usp=sharing ConvNeXt: https://dl.fbaipublicfiles.com/convnext/convnext_base_1k_384.pth
We will update README.md
soon.
missing final_model_csv and pre_model dir while building docker from Dockerfile