Skip to main content

​
prepare_model

Method
This is a method of the TrainVisionModel class in the train_vision module.

​
Signature

def prepare_model() -> Any

​
Returns

​
Returns
Any
The result of the operation.

​
Uses

  • FastVisionModel.from_pretrained
  • FastVisionModel.get_peft_model

​
Used By

​
Source

View on GitHub

praisonai/train_vision.py at line 60