karinaodm

Problem with PyTorch Version 1.10

3

Hi, I am trying to reproduce the results. It works correctly with PyTorch 1.5, but with PyTorch 1.10 - `Parsing Computation Graph with torch.jit failed` and with manual parse_graph function...

Can't use run_segment with apex.amp

4

I use code like this ``` run_segment = optimal_grad_checkpointing(model, inp) run_segment, optimizer = apex.amp.initialize(run_segment, optimizer, opt_level="02", verbosity=0) ... output = run_segment(images) ``` and get the error ``` output = run_segment(images)...

Hellow, thank you for the wonderful work. I converted YOLOv4-tiny model ( [weights](https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights) [cfg](https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov4-tiny.cfg) ) using sh ./prepare_cfg.sh yolov4-tiny.cfg yolov4-tiny_temp.cfg python ./convert_v4.py -n coco.names -c yolov4-tiny_temp.cfg -w yolov4-tiny.weights -m yolov4.mlpackage...