search

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

train using num_steps #752

Closed fangwudi closed 3 years ago

fangwudi commented 3 years ago

❓ Questions and Help

how to train using num_steps instead of num_epochs?

mannatsingh commented 3 years ago

Hi @fangwudi we don't directly support specifying num_steps, but you can tune the number of steps by changing num_epochs (and optionally num_samples in the dataset).

fangwudi commented 3 years ago

Hi @fangwudi we don't directly support specifying num_steps, but you can tune the number of steps by changing num_epochs (and optionally num_samples in the dataset).

thanks, I'm using this method