mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
added debugging util
This commit is contained in:
+2
-1
@@ -47,7 +47,7 @@ def main():
|
||||
model = get_model()
|
||||
|
||||
trainer = Trainer(
|
||||
progress_bar=False,
|
||||
progress_bar=True,
|
||||
experiment=get_exp(),
|
||||
max_nb_epochs=1,
|
||||
train_percent_check=0.4,
|
||||
@@ -66,5 +66,6 @@ def main():
|
||||
|
||||
clear_tt_dir()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user