Skip to content

Instantly share code, notes, and snippets.

@williamFalcon

williamFalcon/cl_2.py

Last active August 3, 2019 12:46
Show Gist options
  • Save williamFalcon/ddb381d271426b7c1c32e712d0f26be9 to your computer and use it in GitHub Desktop.
Save williamFalcon/ddb381d271426b7c1c32e712d0f26be9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cl_2.py
from pytorch_lightning import Trainer
from test_tube import Experiment
model = CoolModel()
exp = Experiment(save_dir=os.getcwd())
# train on cpu using only 10% of the data and limit to 1 epoch (for demo purposes)
trainer = Trainer(experiment=exp, max_nb_epochs=1, train_percent_check=0.1)
trainer.fit(model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment