random

thuml · Sep 21, 2018 · 4c2c105 · 4c2c105
1 parent 14b489c
commit 4c2c105
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch/src/train.py b/pytorch/src/train.py
@@ -283,6 +283,7 @@ def train(config):
             "params":{"vgg_name":args.net, "use_bottleneck":True, "bottleneck_dim":256, "new_cls":True} }
     config["loss"]["random"] = args.random
     config["loss"]["random_dim"] = 1024
+    print(config["loss"]["random"])
 
     config["optimizer"] = {"type":"SGD", "optim_params":{"lr":1.0, "momentum":0.9, \
                            "weight_decay":0.0005, "nesterov":True}, "lr_type":"inv", \