....and added incomplete function (untested)

bouracha · Nov 4, 2020 · 0c85b3d · 0c85b3d
1 parent 70c7309
commit 0c85b3d
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/main.py b/main.py
@@ -45,7 +45,6 @@ def main(opt):
     if opt.dataset=='h3.6m':
       print(">>> validation data {}".format(data.val_dataset.__len__()))
 
-
     # define log csv file
     script_name = os.path.basename(__file__).split('.')[0]
     script_name = script_name + "_{}_in{:d}_out{:d}_dctn{:d}_dropout_{}".format(str(opt.dataset), opt.input_n, opt.output_n, opt.dct_n, str(opt.dropout))
@@ -54,7 +53,6 @@ def main(opt):
     if opt.variational:
         script_name = script_name + "_var_lambda_{}_nz_{}_lr_{}_n_layers_{}".format(str(opt.lambda_), str(opt.n_z), str(opt.lr), str(opt.num_decoder_stage))
 
-
     print(">>> creating model")
     model = nnmodel.GCN(input_feature=dct_n, hidden_feature=opt.linear_size, p_dropout=opt.dropout,
                         num_stage=opt.num_stage, node_n=data.node_n, variational=opt.variational, n_z=opt.n_z, num_decoder_stage=opt.num_decoder_stage)
@@ -64,14 +62,6 @@ def main(opt):
     print(">>> total params: {:.2f}M".format(sum(p.numel() for p in model.parameters()) / 1000000.0))
     methods.optimizer = torch.optim.Adam(model.parameters(), lr=opt.lr)
 
-
-
-
-
-
-
-
-
     for epoch in range(start_epoch, opt.epochs):
 
         if (epoch + 1) % opt.lr_decay == 0: