search

Runinho / pytorch-cutpaste

unoffical and work in progress PyTorch implementation of CutPaste
https://runinho.github.io/pytorch-cutpaste/
229 stars 50 forks source link

--no-pretrained ? #27

Closed danielsoy closed 1 year ago

danielsoy commented 1 year ago

what does the argument --no-pretrained mean, please?

does it train a model from scratch, or is it a kind of transfer learning?

can I use --no-pretrained arg, training with CPU, instead of GPU?

Runinho commented 1 year ago

what does the argument --no-pretrained mean, please?

Check out python run_training.py --help to get a small description for each parameter.

does it train a model from scratch, or is it a kind of transfer learning?

If --no-pretrained is set the model weights of the resnet18 will be randomly initialized. If it is not set the pretrained weights that come with pytorch are used. The relevant line in the code is this one.

can I use --no-pretrained arg, training with CPU, instead of GPU?

Should work, just include --cuda 0 in your arguments.

danielsoy commented 1 year ago

Great!