Update.

[picoclvr.git] / main.py

diff --git a/main.py b/main.py
index 69ee58f..e18887b 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -14,6 +14,7 @@ import torch, torchvision
 from torch import nn
 from torch.nn import functional as F
 
+import ffutils
 import mygpt, tasks
 
 ######################################################################