from torch import nn
from torch.nn import functional as F
+import ffutils
import mygpt, tasks
######################################################################
if args.task == "sandbox":
task = tasks.SandBox(
+ tasks.ProblemByheart(),
nb_train_samples=args.nb_train_samples,
nb_test_samples=args.nb_test_samples,
batch_size=args.batch_size,