Fix failing example

This commit is contained in:
Alexander Engelsberger 2021-05-21 18:54:47 +02:00
parent 2aa631f4e6
commit e7e6bf9173

View File

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