pytorch常用知识点总结.md

pytorch常用知识点总结

---

指定GPU运行指令

# 使用终端命令行运行的 GPU 指定方式
CUDA_VISIBLE_DEVICES=1 python python_script.py
# 在 python 脚本中的 GPU 指定方式
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
# set_device不建议使用
torch.cuda.set_device(id)

os.mkdir()和os.makedirs()区别

• mkdir只创建最后一级目录，mkdirs创建递归目录

Matplotlib中plt.rcParams用法（设置图像细节）

import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

@staticmethod和@classmethod的作用与区别

• 一般来说，要使用类的方法，需先实例化对象再调用其方法，使用@staticmethod或@classmethod，就可以不需要实例化，直接类名.方法名()来调用。
• 从它们的使用上来看,@staticmethod不需要表示自身对象的self和自身类的cls参数，就跟使用函数一样。@classmethod也不需要self参数，但第一个参数需要是表示自身类的cls参数。
• 如果在@staticmethod中要调用到这个类的一些属性方法，只能直接类名.属性名或类名.方法名。而@classmethod因为持有cls参数，可以来调用类的属性，类的方法，实例化对象等，避免硬编码。

class A(object):
    bar = 1
    def foo(self):
        print 'foo'
 
    @staticmethod
    def static_foo():
        print 'static_foo'
        print A.bar
 
    @classmethod
    def class_foo(cls):
        print 'class_foo'
        print cls.bar
        cls().foo()
 
A.static_foo()
A.class_foo()

nn.Upsample