Skip to content

Instantly share code, notes, and snippets.

@THEFASHIONGEEK

THEFASHIONGEEK/Train.py

Last active February 17, 2020 16:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save THEFASHIONGEEK/ba7fa11d26aec7a09f8aa6a1c1480331 to your computer and use it in GitHub Desktop.
Save THEFASHIONGEEK/ba7fa11d26aec7a09f8aa6a1c1480331 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Train.py
import os
import sys
sys.path.append("./monk/")
import psutil
from pytorch_prototype import prototype
ptf = prototype(verbose=1);
ptf.Prototype("sample-project-1", "sample-experiment-1")
ptf.Default(dataset_path="./monk/datasets/train", model_name="resnet18", freeze_base_network=True, num_epochs=10)
ptf.Train()
@THEFASHIONGEEK
Copy link
Author

Train.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment