tmabraham

tmabraham

tmabraham

from timm import create_model
from fastai.vision.learner import _add_norm

class TimmBody(nn.Module):
    def __init__(self, arch:str, pretrained:bool=True, cut=None, n_in:int=3):
        super().__init__()
        model = create_model(arch, pretrained=pretrained, num_classes=0, in_chans=n_in)
        if isinstance(cut, int): self.model = nn.Sequential(*list(model.children())[:cut])
        elif callable(cut): self.model = cut(model)
        elif cut is None: self.model = model

tmabraham

For each function/class please check the following:

• Is every argument on a new line, unless it's a self or cls keyword?
• Is it formatted like this? (Pay attention to the location of the arguments on the newline and the closing parentheses):

def __init__(self, 
    a, 
    b, 
    c
)

tmabraham

import gradio as gr
import fastai
import skimage

learn = load_learner('export.pkl')

labels = learn.dls.vocab
def predict(img):
    img = PILImage.create(img)
    pred,pred_idx,probs = learn.predict(img)

tmabraham

import warnings
warnings.filterwarnings('ignore')

import torch_xla
import torch_xla.distributed.data_parallel as dp
import torch_xla.utils.utils as xu
import torch_xla.core.xla_model as xm
import torch_xla.distributed.parallel_loader as pl
import torch_xla.distributed.xla_multiprocessing as xmp
import torch