model path

[namemute]

Hello author, I have finished running the model, but did not find the generated model file, will the code generate the model file? Such as "xxx.pth"

[zaibrua]

Hello, could you please tell me the configuration of your computer? There will be an error when I use 3090Ti to reproduce, I am not sure if there is a problem with my configuration

[VlSomers]

Hello @namemute, if you want to save the model ".pth" file, you should use the configuration "cfg.model.save_model_flag=True" (Have a look at all available configs inside torchreid/scripts/default_config.py). @zaibrua you should maybe try to use a smaller backbone or a bigger GPU.

[namemute]

Hello, could you please tell me the configuration of your computer? There will be an error when I use 3090Ti to reproduce, I am not sure if there is a problem with my configuration

4090, but I change batch_size from 64 to 32

[namemute]

Hello @namemute, if you want to save the model ".pth" file, you should use the configuration "cfg.model.save_model_flag=True" (Have a look at all available configs inside torchreid/scripts/default_config.py). @zaibrua you should maybe try to use a smaller backbone or a bigger GPU.

thanks