qq_41233323的博客

Pytorch加载模型当在特殊环境下会出现模型无法通过设置pretrained=True的方式来完成自动加载，这时我们需要完成手动加载。我们假设模型设置为：class Model(nn.Module): def __init__(self, num_classes=21): super(Model, self).__init__() self.backbone = timm.create_model('swin_tiny_patch4_window7_224',

DistributedParallelDistributedParallel本身的目标是为了完成多机多卡的分布式训练。当然也可以完成单机多卡训练。运行方式python -m torch.distributed.launch --nproc_per_node=8 YOUR_TRAINING_SCRIPT.py (--arg1 --arg2 --arg3 and all other arguments of your training scrip

TensorFlow2.x中的常用函数合并与分割tf.concat()在特定的维度上进行合并tf.concat(tensor, axis)其中 tensors 保存了所有需要合并的张量 List，axis 指定需要合并的维度。a = tf.random.normal([4,35,8]) b = tf.random.normal([6,35,4])tf.concat([a,b], ax...

MacOS环境下基于Anaconda的Tensorflow安装因为MacOS的环境下，tensorflow暂不支持对于AMD显卡的ROCm驱动，因此安装CPU版本的TensorflowTensorFlow2.x安装（base环境）进入到base虚拟环境下#使用pip安装pip install tensorflow可能会出现run timeout使用镜像安装即可pip instal...