Upload 18 files

models/baseresNet18/best_model_params.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a3620131e9933ba55c1489ad363b4101eb922e04113cc5b53ef77a0fb39811c
+size 44796394

models/baseresNet50/best_model_params.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e22799c79cd7b23cab930ed73e04f1a65f86c2588118359370f92e483fdc3fe
+size 94393946

models/bestresNet50/best_model_params.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e8e5dd4e6152e3cc37d935c9ebdc716c9aaf5096b5e337f60ca126f1b1e2e48
+size 94393946

models/promptMovieBert/config.json

@@ -0,0 +1,39 @@
+{
+  "_name_or_path": "distilbert-base-uncased",
+  "activation": "gelu",
+  "architectures": [
+    "DistilBertForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "dim": 768,
+  "dropout": 0.1,
+  "hidden_dim": 3072,
+  "id2label": {
+    "0": "PG",
+    "1": "PG13",
+    "2": "R",
+    "3": "X",
+    "4": "XXX"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "PG": 0,
+    "PG13": 1,
+    "R": 2,
+    "X": 3,
+    "XXX": 4
+  },
+  "max_position_embeddings": 512,
+  "model_type": "distilbert",
+  "n_heads": 12,
+  "n_layers": 6,
+  "pad_token_id": 0,
+  "problem_type": "single_label_classification",
+  "qa_dropout": 0.1,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": false,
+  "tie_weights_": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.39.3",
+  "vocab_size": 30522
+}

models/promptMovieBert/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30c6642ca63222b3d93a646214d678aa07163aa69d825ac71223598d9538da4a
+size 267841796

models/promptMovieBert/special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

models/promptMovieBert/tokenizer_config.json

@@ -0,0 +1,57 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "never_split": null,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "DistilBertTokenizer",
+  "unk_token": "[UNK]"
+}

models/promptMovieBert/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6fc7cb72b3945fe78a8cd61ce4a8b1b016ac35384f4738a5ac14eaab8bc5a624
+size 4920

models/promptMovieRoberta/config.json

@@ -0,0 +1,43 @@
+{
+  "_name_or_path": "FacebookAI/roberta-large-mnli",
+  "_num_labels": 3,
+  "architectures": [
+    "RobertaForSequenceClassification"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "bos_token_id": 0,
+  "classifier_dropout": null,
+  "eos_token_id": 2,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 1024,
+  "id2label": {
+    "0": "PG",
+    "1": "PG13",
+    "2": "R",
+    "3": "X",
+    "4": "XXX"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 4096,
+  "label2id": {
+    "PG": 0,
+    "PG13": 1,
+    "R": 2,
+    "X": 3,
+    "XXX": 4
+  },
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 514,
+  "model_type": "roberta",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "pad_token_id": 1,
+  "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
+  "torch_dtype": "float32",
+  "transformers_version": "4.39.3",
+  "type_vocab_size": 1,
+  "use_cache": true,
+  "vocab_size": 50265
+}

models/promptMovieRoberta/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fb81cf3b3eb07f17f9922a072e95f983d265a38fc0ee0ae87567bccc612d5886
+size 1421507716

models/promptMovieRoberta/special_tokens_map.json

@@ -0,0 +1,15 @@
+{
+  "bos_token": "<s>",
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "mask_token": {
+    "content": "<mask>",
+    "lstrip": true,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "unk_token": "<unk>"
+}

models/promptMovieRoberta/tokenizer_config.json

@@ -0,0 +1,57 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50264": {
+      "content": "<mask>",
+      "lstrip": true,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "<s>",
+  "eos_token": "</s>",
+  "errors": "replace",
+  "mask_token": "<mask>",
+  "model_max_length": 512,
+  "pad_token": "<pad>",
+  "sep_token": "</s>",
+  "tokenizer_class": "RobertaTokenizer",
+  "trim_offsets": true,
+  "unk_token": "<unk>"
+}

models/promptMovieRoberta/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89214cef336b7fea946c45ddebf2834faa6d5072d1db9c28b9a32eb2700b2c0d
+size 4920

models/readme.md

@@ -0,0 +1,31 @@
+## Dataset
+
+These models are trained on the 13k images dataset from the community images sourced 4/8/24. The 13k images had to have >2 reviewed labels with >50% community agreement. The dataset consists of prompts, tags, and images - although at this time the tags were neglected since there are considerable post-generation additions by the community and moderator, and this data would not be available at the time of scoring. 
+ 
+## Training details
+
+model: baseresNet18
+- epochs: 39
+- stopped due to early stopping
+- train_test_split: .85/.15
+-  augmentations: None
+
+model: baseresNet50
+- epochs: 39
+- stopped due to early stopping
+- train_test_split: .85/.15
+- augmentations: None
+
+model: bestresNet50
+- epochs: 25
+- stopped due to max epochs
+- train_test_split: .8/.2
+- augmentations: RandomRotation, RandomHorizontalFlip, RandomResizedCrop
+
+model: promptMovieBert
+- epochs: 10
+- test_train_split: .9/.1
+
+model: promptMovieRoberta
+- epochs: 10
+- test_train_split: .9/.1