linksense / lightnetplusplus Goto Github PK

...
> # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ #
> 1. Setting up Model...
Traceback (most recent call last):
  File "deploy/cityscapes/evaluation/ss.py", line 253, in <module>
    model.load_state_dict(pre_weight)
  File "/home/ben/LightNetPlusPlus/env/lib/python3.6/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DataParallel:
	Missing key(s) in state_dict: "module.usm.norm_act.weight", "module.usm.norm_act.bias", "module.usm.norm_act.running_mean", "module.usm.norm_act.running_var". 

(env) ben@mercury:~/LightNetPlusPlus$ grep -r norm_act.weight ./*
Binary file ./checkpoint/cityscapes_mobilenetv2plus_x1.0.pkl matches
Binary file ./checkpoint/cityscapes_shufflenetv2plus_x0.5.pkl matches
Binary file ./checkpoint/cityscapes_shufflenetv2plus_x1.0.pkl matches
Binary file ./checkpoint/deepdrive_shufflenetv2plus_x1.0.pkl matches

(env) ben@mercury:~/LightNetPlusPlus$ grep -r usm.norm_act.weight ./*
Binary file ./checkpoint/deepdrive_shufflenetv2plus_x1.0.pkl matches

Traceback (most recent call last):    
  File "/home/wcgu/code/RTSeg/LightNetPlusPlus/scripts/train_shuffle.py", line 382, in <module>    
    train(train_args, data_path, save_path)    
  File "/home/wcgu/code/RTSeg/LightNetPlusPlus/scripts/train_shuffle.py", line 290, in train    
    val_loss = loss_fn(input=net_out, target=labels_val, K=topk, weight=None)    
  File "/home/wcgu/code/RTSeg/LightNetPlusPlus/scripts/loss.py", line 111, in bootstrapped_cross_entropy2d    
    size_average=size_average)    
  File "/home/wcgu/code/RTSeg/LightNetPlusPlus/scripts/loss.py", line 88, in _bootstrap_xentropy_single    
    log_p = log_p[target.view(n * h * w, 1).repeat(1, c) >= 0]   
RuntimeError: CUDA out of memory. Tried to allocate 304.00 MiB (GPU 0; 10.72 GiB total capacity; 9.30 GiB already allocated; 226.62 MiB free; 59.54 MiB cached)