HRNet-Image-Classification

HRNet-Image-Classification copied to clipboard

How to train on single GPU

4

For I don't have multi-GPUS, I changed the GPUS = (0,0,0,0), but this error came `RuntimeError: inputs must be on unique devices.` There is no doubt telling that this network...

Chen65010445


wywywy01

Torch.jit.script not working with this model

2

I have tried to convert this model to TorchScript using torch.jit.script. However, I am getting this issue: ``` RuntimeError: Expected integer literal for index: File "/home/david/Documents/pry/models/archs/hrnet.py", line 228 for i...

WaterKnight1998

the loss become nan

1

when i was trying to recurrence the face-xray, I modified the HRNet-Image-Calssification, but I got a bug that loss is nan. this is what i added after stage4 in the...

niyunsheng

It doesn't work

3


gunerhui

Thanks for such a great work. I trained HRNet-W32-C in imagenet, got the pretrained cls model (final_state.pth.tar) which has 1956 keys. However, the pretrained model (hrnet_w32-36af842e.pth) provided by pose_hrnet_w32 [https://github.com/leoxiaobin/deep-high-resolution-net.pytorch]...

rujiewu

This code is very unfriendly to windows, and it took me two hours to make it run on windows.

1

douyimin

A solution: HRNet Backbone Adopt_different_blocks_bug(BASIC//BOTTLENECK)

2

https://github.com/HRNet/HRNet-Image-Classification/blob/8f158719e821836e21e6cba99a3241a12a13bc41/lib/models/cls_hrnet.py#L459~L473 If different block types are used in different stages, instead of the default bottleneck-basic-basic-basic in the original yaml file, the channel mismatch error as shown in the figure below...

Taylor-X76

guide to use pre-trained model

1

Hello. I am a beginner in deep learning and pytorch, and I have a question about how to use HRNet. I downloaded the pre-trained model you provided (HRNet-W64-C) from one-drive...

doljae

HI! from ~\lib\datasets\cityscapes.py i can see: self.class_weights = torch.FloatTensor([0.8373, 0.918, 0.866, 1.0345, 1.0166, 0.9969, 0.9754, 1.0489, 0.8786, 1.0023, 0.9539, 0.9843, 1.1116, 0.9037, 1.0865, 1.0955, 1.0865, 1.1529, 1.0507]).cuda() What do you...

yuanry16