ShawnLiu1011

Initialization for premodel of different nets

9

Traceback (most recent call last): File "./tools/trainval_net.py", line 150, in max_iters=args.max_iters) File "/home/nieqinqin/liuxiaoyu/CLWS/tools/../lib/model/train_val.py", line 365, in train_net sw.train_model(max_iters) File "/home/nieqinqin/liuxiaoyu/CLWS/tools/../lib/model/train_val.py", line 239, in train_model lr, last_snapshot_iter, stepsizes, np_paths, ss_paths =...

memory error when train voc dataset

2

``` Loading initial model weights from data/imagenet_weights/res101.pth Loaded. Traceback (most recent call last): File "./tools/trainval_net.py", line 192, in sw.train_model(iters_sum) File "/home/nieqinqin/liuxiaoyu/SSM-Pytorch/tools/../lib/model/train_val.py", line 295, in train_model self.net.train_step_with_summary(blobs, self.optimizer) File "/home/nieqinqin/liuxiaoyu/SSM-Pytorch/tools/../lib/nets/network.py", line...