Improve example test script (with failing example)

This commit is contained in:
Alexander Engelsberger
2021-05-21 18:48:37 +02:00
parent c6992da123
commit 2aa631f4e6
3 changed files with 15 additions and 5 deletions

View File

@@ -4,7 +4,8 @@ import argparse
import prototorch as pt
import pytorch_lightning as pl
import torch
#import torch
if __name__ == "__main__":
# Command-line arguments