-
I have a bit of a doubt about this code of base_train.py. I think the location of some codes should be reversed, right?
for name, param in model.named_parameters():
for ft_layer in p['…
-
**Describe the issue**:
I was trying to prune this custom model of mine
```
class ResNetArc_Classifier(nn.Module):
def __init__(self, model_type:str="resnet50", class_count:int=1000):
…
-
# Prerequisites
Please answer the following questions for yourself before submitting an issue.
- [ x ] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
- [ x ] I am repor…
-
Click to expand!
### Issue Type
Feature Request
### Have you reproduced the bug with TF nightly?
Yes
### Source
source
### Tensorflow Version
TensorFlow version 2.13.0-dev20230215
### Cust…
-
Hello, I would like to follow this issue #418 up, since it is closed and not answered in detailed .
Indeed, the output of the resnet encoder is a list containing the intermediate outputs of all the…
-
Hi all,
I have two issues to report with running working with 4band 16 bit imagery in version 6852729 using a config file similar to https://github.com/NRCan/geo-deep-learning/blob/v.1.2.0/conf/dev…
-
Hello.
I am trying to test the COS.ckpt model on miniImageNet test data, the link to which is provided in the readme file. I have made the required changes in the `set_config_COSOC.py` as mentioned…
-
I used the ROI pooling layer only, but during the training process, the usuage of GPU memory is growing until OOM.
the `nvidia-smi` output
iter 10(appro)
+---------------------------------+------…
-
> kubesphere: `v3.1.1` ,docker: `v20.10.9`,centos7.9
![image](https://user-images.githubusercontent.com/71524776/232651297-5948d1a1-c384-4082-8dd6-8af0e53e142e.png)
`log`:
```shell
[root@ma…
-
### Bug description
I clone and install torchcam,and then I use:
```
!python torch-cam/scripts/cam_example.py \
--img test_img/border-collie.jpg \
--savefig output/B1_border_col…